小程序收集箱:标记特定颜色区域,色彩阀值化处理法(opencv库python语言)

色彩阀值化处理——利用openCV-python中inRange()等相关函数,通过颜色标记代号选取不同的颜色。

#标记特定颜色区域
def Mark_Color(img, flag_color):
    #自定义
    if flag_color ==0:
        lower = np.array([24, 43, 46], dtype="uint8")
        upper = np.array([43, 255, 255], dtype="uint8")
    #黑色
    elif flag_color ==1:
        lower = np.array([0, 0, 0], dtype="uint8")
        upper = np.array([180, 255, 46], dtype="uint8")
    # 灰色
    elif flag_color == 2:
        lower = np.array([0, 0, 46], dtype="uint8")
        upper = np.array([180, 43, 220], dtype="uint8")
    # 白色
    elif flag_color == 3:
        lower = np.array([0, 0, 221], dtype="uint8")
        upper = np.array([180, 30, 255], dtype="uint8")
    # 红色
    elif flag_color == 4:
        lower = np.array([156, 43, 46], dtype="uint8")
        upper = np.array([180, 255, 255], dtype="uint8")
    # 红色2
    elif flag_color == 5:
        lower = np.array([0, 43, 46], dtype="uint8")
        upper = np.array([10, 255, 255], dtype="uint8")
    # 橙色
    elif flag_color == 6:
        lower = np.array([11, 43, 46], dtype="uint8")
        upper = np.array([25, 255, 255], dtype="uint8")
    # 黄色
    elif flag_color == 7:
        lower = np.array([26, 43, 46], dtype="uint8")
        upper = np.array([34, 255, 255], dtype="uint8")
    # 绿色
    elif flag_color == 8:
        lower = np.array([35, 43, 46], dtype="uint8")
        upper = np.array([77, 255, 255], dtype="uint8")
    # 青色
    elif flag_color == 9:
        lower = np.array([78, 43, 46], dtype="uint8")
        upper = np.array([99, 255, 255], dtype="uint8")
    # 蓝色
    elif flag_color == 10:
        lower = np.array([100, 43, 46], dtype="uint8")
        upper = np.array([124, 255, 255], dtype="uint8")
    # 紫色
    elif flag_color == 11:
        lower = np.array([125, 43, 46], dtype="uint8")
        upper = np.array([155, 255, 255], dtype="uint8")

    converted = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
    markingMask = cv2.inRange(converted, lower, upper)
    markingMask = cv2.GaussianBlur(markingMask, (5, 5), 0)
    marked_img = cv2.bitwise_and(img, img, mask=markingMask)
    return marked_img

博文根据颜色不同进行匹配提到多种颜色匹配的方法,请根据需要查阅。

你可能感兴趣的:(计算机视觉,小程序收集箱)