python实现简单颜色识别程序

时间:2021-05-22

本文实例为大家分享了python实现简单颜色识别程序的具体代码,供大家参考,具体内容如下

import numpy as npimport cv2font= cv2.FONT_HERSHEY_SIMPLEXlower_red=np.array([0,127,128])#红色阈值下界higher_red=np.array([10,255,255])#红色阈值上界lower_green=np.array([35,110,106])#绿色阈值下界higher_green=np.array([77,255,255])#绿色阈值上界cap=cv2.VideoCapture(0)#打开电脑内置摄像头if(cap.isOpened()): while(True): ret,frame=cap.read()#按帧读取,这是读取一帧 img_hsv=cv2.cvtColor(frame,cv2.COLOR_BGR2HSV) mask_red=cv2.inRange(img_hsv,lower_red,higher_red)#可以认为是过滤出红色部分,获得红色的掩膜 mask_green=cv2.inRange(img_hsv,lower_green,higher_green)#获得绿色部分掩膜 mask_green = cv2.medianBlur(mask_green, 7) # 中值滤波 mask_red = cv2.medianBlur(mask_red, 7) # 中值滤波 mask=cv2.bitwise_or(mask_green,mask_red)#三部分掩膜进行按位或运算 image1,cnts1,hierarchy1=cv2.findContours(mask_red,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_NONE)#轮廓检测 image3,cnts3,hierarchy3=cv2.findContours(mask_green,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_NONE) for cnt in cnts1: (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) for cnt in cnts3: (x, y, w, h) = cv2.boundingRect(cnt) # 该函数返回矩阵四个点 cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2) # 将检测到的颜色框起来 cv2.putText(frame, 'green', (x, y - 5), font, 0.7, (0,255,0), 2) cv2.imshow('frame',frame) k=cv2.waitKey(20)&0xFF if k ==27: breakcv2.waitKey(0)cv2.destroyAllWindows()

我发现只要多于两种颜色识别起来误差就极大极大,等以后学到会回来进行优化的。目前识别两种颜色还是比较稳的。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章