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(发邮件时,请将#更换为@)进行举报,并提供相关证据,一经查实,本站将立刻删除涉嫌侵权内容。