python利用opencv实现颜色检测
本文实例为大家分享了python利用opencv实现颜色检测的具体代码,供大家参考,具体内容如下
需要实现倒车辅助标记检测的功能,倒车辅助标记颜色已经确定了,所以不需要使用深度学习的方法,那样成本太高了,直接可以使用颜色检测的方法。
1.首先需要确定待检测目标的HSV值
importcv2 img=cv2.imread('l3.png') gray=cv2.cvtColor(img,cv2.COLOR_BGR2GRAY) hsv=cv2.cvtColor(img,cv2.COLOR_BGR2HSV) defmouse_click(event,x,y,flags,para): ifevent==cv2.EVENT_LBUTTONDOWN:#左边鼠标点击 print('PIX:',x,y) print("BGR:",img[y,x]) print("GRAY:",gray[y,x]) print("HSV:",hsv[y,x]) if__name__=='__main__': cv2.namedWindow("img") cv2.setMouseCallback("img",mouse_click) whileTrue: cv2.imshow('img',img) ifcv2.waitKey()==ord('q'): break cv2.destroyAllWindows()
2.然后利用颜色检测,检测出指定目标
importnumpyasnp importcv2 font=cv2.FONT_HERSHEY_SIMPLEX lower_red=np.array([0,127,128])#红色阈值下界 higher_red=np.array([10,255,255])#红色阈值上界 lower_yellow=np.array([15,230,230])#黄色阈值下界 higher_yellow=np.array([35,255,255])#黄色阈值上界 lower_blue=np.array([85,240,140]) higher_blue=np.array([100,255,165]) frame=cv2.imread("l3.png") img_hsv=cv2.cvtColor(frame,cv2.COLOR_BGR2HSV) mask_red=cv2.inRange(img_hsv,lower_red,higher_red)#可以认为是过滤出红色部分,获得红色的掩膜 mask_yellow=cv2.inRange(img_hsv,lower_yellow,higher_yellow)#获得绿色部分掩膜 mask_yellow=cv2.medianBlur(mask_yellow,7)#中值滤波 mask_red=cv2.medianBlur(mask_red,7)#中值滤波 mask_blue=cv2.inRange(img_hsv,lower_blue,higher_blue)#获得绿色部分掩膜 mask_blue=cv2.medianBlur(mask_blue,7)#中值滤波 #mask=cv2.bitwise_or(mask_green,mask_red)#三部分掩膜进行按位或运算 print(mask_red) cnts1,hierarchy1=cv2.findContours(mask_red,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_NONE)#轮廓检测#红色 cnts2,hierarchy2=cv2.findContours(mask_blue,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_NONE)#轮廓检测#红色 cnts3,hierarchy3=cv2.findContours(mask_yellow,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_NONE) forcntincnts1: (x,y,w,h)=cv2.boundingRect(cnt)#该函数返回矩阵四个点 cv2.rectangle(frame,(x,y),(x+w,y+h),(0,0,255),2)#将检测到的颜色框起来 cv2.putText(frame,'red',(x,y-5),font,0.7,(0,0,255),2) forcntincnts2: (x,y,w,h)=cv2.boundingRect(cnt)#该函数返回矩阵四个点 cv2.rectangle(frame,(x,y),(x+w,y+h),(0,0,255),2)#将检测到的颜色框起来 cv2.putText(frame,'blue',(x,y-5),font,0.7,(0,0,255),2) forcntincnts3: (x,y,w,h)=cv2.boundingRect(cnt)#该函数返回矩阵四个点 cv2.rectangle(frame,(x,y),(x+w,y+h),(0,255,0),2)#将检测到的颜色框起来 cv2.putText(frame,'yellow',(x,y-5),font,0.7,(0,255,0),2) cv2.imshow('frame',frame) cv2.waitKey(0) cv2.destroyAllWindows()
3.效果
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持毛票票。
声明:本文内容来源于网络,版权归原作者所有,内容由互联网用户自发贡献自行上传,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任。如果您发现有涉嫌版权的内容,欢迎发送邮件至:czq8825#qq.com(发邮件时,请将#更换为@)进行举报,并提供相关证据,一经查实,本站将立刻删除涉嫌侵权内容。