python+opencv识别魔方颜色+kociemba算法应用+显示识别效果

具体介绍见：https://blog.csdn.net/qq_32107283/article/details/88307423 colordraw.py用来画出由识别后得到的字符串来转换为颜色图，colordetect.py用来实现识别颜色加kociemba算法应用 压缩包里还有对应的六个面的魔方照片，运行python colordetect就可以看效果了 希望有什么建议随时提出谢谢共享～～～

# -*- coding:utf-8 -*-
__author__ = ‘zhanghongye‘
import cv2
import numpy as np
import sys
#sys.path.append（‘/home/zhy/桌面/CubeSolver-master‘）
from colordraw import *
import math
import json
import kociemba
import time
from multiprocessing import Pool
kernel_15 = np.ones（（1515）np.uint8）#15x15的卷积核
kernel_50 = np.ones（（5050）np.uint8）#50x50的卷积核
draw = np.zeros（（48006400 3） dtype=“uint8“）#创建一个高4800*宽6400画布
#处理图片
def colorMatch（side）:
    
    cube_rgb = cv2.imread（ side + ‘.jpg‘）
    cube_gray = cv2.cvtColor（cube_rgb cv2.COLOR_BGR2GRAY）#颜色转换gray
    cube_hsv = cv2.cvtColor（cube_rgbcv2.COLOR_BGR2HSV）#颜色转换hsv
    cube_gray = cv2.adaptiveThreshold（cube_gray255cv2.ADAPTIVE_THRESH_GAUSSIAN_Ccv2.THRESH_BINARY112）#自适应滤波
    #cv2.namedWindow（sidecv2.WINDOW_NORMAL）
    #cv2.imshow（sidecube_rgb）
    # 白色
    lower_white = np.array（[0 0 201]）
    upper_white = np.array（[180 50 255]）
    white_mask = cv2.inRange（cube_hsv lower_white upper_white）
    white_erosion = cv2.erode（white_mask kernel_15 iterations = 1）
    white_res = cv2.bitwise_and（cube_rgb cube_rgb mask = white_erosion）
    #红色
    ‘‘‘lower_red = np.array（[05050]）
    upper_red = np.array（[10255255]）
    red_mask0 = cv2.inRange（cube_hsv lower_red upper_red）
    lower_red = np.array（[172 135 150]）
    upper_red = np.array（[179 240 255]）
    red_mask1 = cv2.inRange（cube_hsv lower_red upper_red）
    red_mask = red_mask0 + red_mask1
    ‘‘‘
    lower_red = np.array（[170 110 145]）
    upper_red = np.array（[182 240 255]）
    red_mask = cv2.inRange（cube_hsv lower_red upper_red）
    red_erosion = cv2.erode（red_mask kernel_15 iterations = 1）
    red_res = cv2.bitwise_and（cube_rgb cube_rgb mask = red_erosion）
    #橙色
    lower_orange = np.array（[3 115 195]）
    upper_orange = np.array（[9 190 255]）
    orange_mask = cv2.inRange（cube_hsv lower_orange upper_orange）
    orange_erosion = cv2.erode（orange_mask kernel_15 iterations = 1）
    orange_res = cv2.bitwise_and（cube_rgb cube_rgb mask = orange_erosion）

    #黄色
    lower_yellow = np.array（[20 125 142]）
    upper_yellow = np.array（[34 243 255]）
    yellow_mask = cv2.inRange（cube_hsv lower_yellow upper_yellow）
    yellow_erosion = cv2.erode（yellow_mask kernel_15 iterations = 1）
    yellow_res = cv2.bitwise_and（cube_rgb cube_rgb mask = yellow_erosion）
    #绿色
    lower_green = np.array（[68140120]）
    upper_green = np.array（[82255245]）
    green_mask = cv2.inRange（cube_hsv lower_green upper_green）
    green_erosion = cv2.erode（green_mask kernel_15 iterations = 1）
    green_res = cv2.bitwise_and（cube_rgb cube_rgb mask = green_erosion）
    #蓝色
    lower_blue = np.array（[95 123 109]）
    upper_blue = np.array（[124 253 240]）
    blue_mask = cv2.inRange（cube_hsv lower_blue upper_blue）
    blue_erosion = cv2.erode（blue_mask kernel_15 iterations = 1）
    blue_res = cv2.bitwise_and（cube_rgb cube_rgb mask = blue_erosion）
    #总掩膜
    mask = red_erosion + green_erosion + yellow_erosion +