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

最新推荐文章于 2024-01-17 20:37:17 发布

原创

最新推荐文章于 2024-01-17 20:37:17 发布 · 1.5k 阅读

18 ·

CC 4.0 BY-SA版权

文章标签：

#python #opencv #图像识别 #机器学习

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

#本程序的功能是识别魔方方格中的颜色
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') #定义图片位置
#img= cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) #转化为