一般主题:通过分类特征提取进行颜色匹配应用
我决定使用" Color Histogram"和" K最近邻"用于分类颜色。
我可以获得图像颜色直方图,我可以将它们部署在带有标签的csv文件中以训练KNN。这一步完成了!但是,我无法弄清楚如何将测试图像颜色直方图值发送给KNN进行分类呢?测试图像颜色直方图中有一堆矩阵值。所以我的问题是如何将这个点云(测试图像的颜色直方图值)发送给KNN进行分类呢?
答案 0 :(得分:0)
您可以将RGB颜色空间转换为HSV颜色空间,并使用Hue通道及其标签作为KNN分类器功能的训练输入。以下是一个简单的实现:
hsv_img = cv2.cvtColor(rgbimage,cv2.COLOR_BGR2HSV)
hue = hsv_img[:,:,:1].flatten()