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
python识别红绿灯
最新推荐文章于 2024-07-18 23:33:05 发布