Opencv-Python(9)-视频-MASK-颜色检测

屏幕里只显示一种颜色。有4个屏幕，右上角为原始图像。
左上，左下，右下的视窗分别定义为绿色、橘色、紫色。
当原始视频中出现了这个颜色，相应的屏幕会出现视图。

import cv2
import numpy as np

frameWidth = 640
frameHeight = 480
cap = cv2.VideoCapture(0)
cap.set(3,frameWidth)
cap.set(4,frameHeight)
cap.set(10,130)


myColors = [[5,107,0,19,255,255],
           [133,56,0,159,156,255],
           [57,76,0,100,255,255]]

# 找一种颜色
def findColor(img,myColors):
    imgHSV = cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
    lower = np.array(myColors[0][0:3])
    upper = np.array(myColors[0][3:6])
    mask = cv2.inRange(imgHSV, lower, upper)
    cv2.imshow("img",mask)

# 开不同的窗口显示不同的颜色
def findColors(img,myColors):
    imgHSV= cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
    count =0
    for color in myColors:
        lower = np.array(color[0:3])
        upper = np.array(color[3:6])
        mask = cv2.inRange(imgHSV, lower, upper)
        cv2.imshow(str(color[0]), mask)

while True:
    success,img = cap.read()
    findColors(img,myColors)
    cv2.imshow("result",img)
    if cv2.waitKey(1)&0XFF == ord('q'):
        break