#https://blog.csdn.net/clickyeah/article/details/51750199
import cv2
import numpy as np
import sys
# src_cv_img = cv2.imread(sys.argv[1])
src_cv_img = cv2.imread('zjz.jpg')

h,w,c = src_cv_img.shape
print(h,w,c)
cv_img = src_cv_img.reshape(-1,3)
print(cv_img.shape)
from sklearn.cluster import KMeans
estimator = KMeans(n_clusters=4)#构造聚类器
estimator.fit(cv_img)
label_pred = estimator.labels_
x = np.argmax(np.bincount(label_pred))
for i,j in enumerate(cv_img):
    if(label_pred[i] == x):
        cv_img[i] = np.asarray([0,0,255])

cv2.imwrite("tmp4.jpg",cv_img.reshape(h,w,c))
# cv2.imshow('result',cv_img)
# cv2.waitKey(0)
# cv2.destroyAllWindows()