魔方机器人之SVM算法识别魔方颜色

Crack_C

于 2021-06-18 20:36:44 发布

阅读量1.3k

点赞数

文章标签： python opencv 图像识别机器学习

本文链接：https://blog.csdn.net/Crack_C/article/details/118033419

版权

本Python程序通过OpenCV库实现SVM算法训练，用于识别魔方方格的颜色。程序加载训练数据，设置SVM参数并进行训练，然后读取图片并实时检测鼠标移动时的像素颜色，根据SVM预测结果输出对应的颜色（绿色、蓝色、白色、黄色、棕色、红色或其他）。

摘要由CSDN通过智能技术生成

#本程序的功能是识别魔方方格中的颜色
import cv2
import numpy as np

# Set up training data
## [setup1]
labels = np.array([1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 0, 0])#绿，蓝，白，黄，棕，红，其他
trainingData = np.matrix([[86,133,35],[58,103,6],[210,151,125],[110,51,25],[216,214,206],[209,207,199],[89,193,206],[54,158,171],[57,91,174],[43,77,160],[55,52,131],[28,25,104],[160,173,181],[27,25,24]], dtype=np.float32)
## [setup1]

# Train the SVM
## [init]
svm = cv2.ml.SVM_create()
svm.setType(cv2.ml.SVM_C_SVC)
svm.setKernel(cv2.ml.SVM_LINEAR)
svm.setTermCriteria((cv2.TERM_CRITERIA_MAX_ITER, 100, 1e-6))
## [init]
## [train]
svm.train(trainingData, cv2.ml.ROW_SAMPLE, labels)
## [train]

img= cv2.imread('cube.jpg'