基于opencv图像处理对交通路口的红绿灯进行颜色检测( 二 )

运行代码后调整的结果(如下图所示) , 很明显可以看到绿色已经被获取到 。
基于opencv图像处理对交通路口的红绿灯进行颜色检测文章插图
4.二值图像颜色判定因为图像是二值的图像 , 所以如果图像出现白点 , 也就是255 , 那么就取他的max最大值255 , 视频帧的不断变化然后遍历每个颜色值
red_color = np.max(red_blur)green_color = np.max(green_blur)if red_color == 255: print('red')elif green_color == 255: print('green')5.颜色结果画在图像上用矩形框来框选出红绿灯区域
cv2.rectangle(frame,(1020,50),(1060,90),(0,0,255),2) #按坐标画出矩形框cv2.putText(frame, "red", (1020, 40), cv2.FONT_HERSHEY_COMPLEX, 1, (0, 0, 255),2)#显示red文本信息6.完整代码import cv2import numpy as npcap = cv2.VideoCapture('video/小路口.mp4')while True:ret,frame = cap.read()if ret == False:breakframe = cv2.resize(frame,(1920,1080))#截取roi区域roiColor = frame[50:90,950:1100]#转换hsv颜色空间hsv = cv2.cvtColor(roiColor,cv2.COLOR_BGR2HSV)#redlower_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)#greenlower_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最大值255red_color = np.max(red_blur)green_color = np.max(green_blur)#在red_color中判断二值图像如果数值等于255 , 那么就判定为redif red_color == 255:print('red')# 。。。 这是我经常会混淆的坐标 。。。就列举出来记一下 。。。#yy+h xx+w#frame[50:90,950:1100]#xyx+wy+hcv2.rectangle(frame,(1020,50),(1060,90),(0,0,255),2) #按坐标画出矩形框cv2.putText(frame, "red", (1020, 40), cv2.FONT_HERSHEY_COMPLEX, 1, (0, 0, 255),2)#显示red文本信息#在green_color中判断二值图像如果数值等于255 , 那么就判定为greenelif green_color == 255:print('green')cv2.rectangle(frame,(1020,50),(1060,90),(0,255,0),2)cv2.putText(frame, "green", (1020, 40), cv2.FONT_HERSHEY_COMPLEX, 1, (0, 255, 0),2)cv2.imshow('frame',frame)red_blur = cv2.resize(red_blur,(300,200))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检测红灯的效果(如下图所示)
基于opencv图像处理对交通路口的红绿灯进行颜色检测文章插图
检测绿灯的效果(如下图所示)
基于opencv图像处理对交通路口的红绿灯进行颜色检测文章插图
最后!!!完整项目代码获取后台私信小编01即可
第一次接触opencv!所以请各位视觉领域的大佬们勿喷我这个小菜鸡!(/狗头) 代码量非常少 , 无泛化能力 , 很low的一种做法 。。。 不过对于小白的我来说学习hsv颜色空间还是很有帮助滴!干就完了!奥利给!
大家可以多多支持一下小明童鞋!
此文转载文 , 著作权归作者所有 , 如有侵权联系小编删除!
【基于opencv图像处理对交通路口的红绿灯进行颜色检测】


推荐阅读