基于KNN算法的颜色识别

KNN实现魔方颜色识别

1.颜色识别

该程序利用KNN实现魔方颜色识别,特征采用RGB的颜色直方图,它可以区分白、黑、红、绿、蓝、橙色、黄色和紫色,如果你想分类更多颜色或者提高分类准确率,可以通过调整训练集数据或者考虑使用其他的特征如颜色矩)或者颜色相关图。

你可以使用color_recognition_api实现实时的颜色识别,作者提供了两种识别方式,一种是摄像头实时识别颜色(color_classification_webcam.py),

基于KNN算法的颜色识别_第1张图片

另一种实现对单张图片的颜色识别(color_classification_image.py )

基于KNN算法的颜色识别_第2张图片

该程序实现的功能:

  • 特征提取:对训练集图像提取RGB颜色直方图特征。

  • 训练KNN分类器:通过RGB颜色直方图值训练KNN分类器

  • 通过训练好的KNN分类器分类:逐帧读取摄像头,对每帧图像进行特征提取,最后通过训练好的KNN分类器判别物体的颜色。

TODS:

你可能感兴趣的:(大数据及数据管理(治理)专栏,算法,计算机视觉,人工智能)