基于Kmeans的证件照背景色替换算法-添加python实现

最新推荐文章于 2024-09-14 09:14:02 发布

置顶 clickyeah

最新推荐文章于 2024-09-14 09:14:02 发布

阅读量4.8k

点赞数 2

分类专栏：算法文章标签：算法 kmeans 换背景颜色 python

本文链接：https://blog.csdn.net/clickyeah/article/details/51750199

版权

算法专栏收录该内容

17 篇文章 0 订阅

订阅专栏

import cv2
import numpy as np
import sys
src_cv_img = cv2.imread(sys.argv[1])
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([255,255,255])

output_cv_img = cv_img.reshape(h,w,c)

cv2.imwrite("tmp.jpg",output_cv_img)

原照片：