屏幕里只显示一种颜色。有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