import cv2
import numpy as np
cap = cv2.VideoCapture('红绿灯.mp4')
while True:
ret,frame = cap.read()
if ret == False:
break
frame = cv2.resize(frame,(1620,900))
#截取roi区域
roiColor = frame[100:500,300:800]
#转换hsv颜色空间
hsv = cv2.cvtColor(roiColor,cv2.COLOR_BGR2HSV)
#red
lower_hsv_red = np.array([157,177,122])#红上
upper_hsv_red = np.array([179,255,255])#红下
mask_red = cv2.inRange(hsv,lowerb=lower_hsv_red,upperb=upper_hsv_red)#过滤红色部分
#中值滤波
red_blur = cv2.medianBlur(mask_red, 7)
#green
lower_hsv_green = np.array([49,79,137])
upper_hsv_green = np.array([90,255,255])
mask_green = cv2.inRange(hsv,lowerb=lower_hsv_green,upperb=upper_hsv_green)
#中值滤波
green_blur = cv2.medianBlur(mask_green, 7)
#因为图像是二值的图像,所以如果图像出现白点,也就是255,那么就取他的max最大值255
red_color = np.max(red_blur)
green_color = np.max(green_blur)
#在red_color中判断二值图像如果数值等于255,那么就判定为red
if red_color == 255:
print('red')
cv2.rectangle(frame,(660,420),(400,300),(0,0,255),2) #按坐标画出矩形框
cv2.putText(frame, "red", (500, 280), cv2.FONT_HERSHEY_COMPLEX, 1, (0, 0, 255),2)#显示red文本信息
#在green_color中判断二值图像如果数值等于255,那么就判定为green
elif green_color == 255:
print('green')
cv2.rectangle(frame, (660, 420), (400, 300), (0, 0, 255), 2)
cv2.putText(frame, "green", (500, 280), cv2.FONT_HERSHEY_COMPLEX, 1, (0, 255, 0),2)
cv2.imshow('frame',frame)
red_blur = cv2.resize(red_blur,(200,150))
green_blur = cv2.resize(green_blur,(300,200))
cv2.imshow('red_window',red_blur)
cv2.imshow('green_window',green_blur)
c = cv2.waitKey(10)
if c==27:
break
kaishiopenCV识别红绿灯
最新推荐文章于 2024-06-23 15:20:35 发布