当前位置:   article > 正文

python opencv检测目标颜色_python opencv轮廓区域颜色

python opencv轮廓区域颜色
  1. # -*- coding:utf-8 -*-
  2. __author__ = 'kingking'
  3. __version__ = '1.0'
  4. __date__ = '14/07/2017'
  5. import cv2
  6. import numpy as np
  7. import time
  8. if __name__ == '__main__':
  9. Img = cv2.imread('example.png')#读入一幅图像
  10. kernel_2 = np.ones((2,2),np.uint8)#2x2的卷积核
  11. kernel_3 = np.ones((3,3),np.uint8)#3x3的卷积核
  12. kernel_4 = np.ones((4,4),np.uint8)#4x4的卷积核
  13. if Img is not None:#判断图片是否读入
  14. HSV = cv2.cvtColor(Img, cv2.COLOR_BGR2HSV)#把BGR图像转换为HSV格式
  15. '''
  16. HSV模型中颜色的参数分别是:色调(H),饱和度(S),明度(V)
  17. 下面两个值是要识别的颜色范围
  18. '''
  19. Lower = np.array([20, 20, 20])#要识别颜色的下限
  20. Upper = np.array([30, 255, 255])#要识别的颜色的上限
  21. #mask是把HSV图片中在颜色范围内的区域变成白色,其他区域变成黑色
  22. mask = cv2.inRange(HSV, Lower, Upper)
  23. #下面四行是用卷积进行滤波
  24. erosion = cv2.erode(mask,kernel_4,iterations = 1)
  25. erosion = cv2.erode(erosion,kernel_4,iterations = 1)
  26. dilation = cv2.dilate(erosion,kernel_4,iterations = 1)
  27. dilation = cv2.dilate(dilation,kernel_4,iterations = 1)
  28. #target是把原图中的非目标颜色区域去掉剩下的图像
  29. target = cv2.bitwise_and(Img, Img, mask=dilation)
  30. #将滤波后的图像变成二值图像放在binary中
  31. ret, binary = cv2.threshold(dilation,127,255,cv2.THRESH_BINARY)
  32. #在binary中发现轮廓,轮廓按照面积从小到大排列
  33. contours, hierarchy = cv2.findContours(binary,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)
  34. p=0
  35. for i in contours:#遍历所有的轮廓
  36. x,y,w,h = cv2.boundingRect(i)#将轮廓分解为识别对象的左上角坐标和宽、高
  37. #在图像上画上矩形(图片、左上角坐标、右下角坐标、颜色、线条宽度)
  38. cv2.rectangle(Img,(x,y),(x+w,y+h),(0,255,),3)
  39. #给识别对象写上标号
  40. font=cv2.FONT_HERSHEY_SIMPLEX
  41. cv2.putText(Img,str(p),(x-10,y+10), font, 1,(0,0,255),2)#加减10是调整字符位置
  42. p +=1
  43. print '黄色方块的数量是',p,'个'#终端输出目标数量
  44. cv2.imshow('target', target)
  45. cv2.imshow('Mask', mask)
  46. cv2.imshow("prod", dilation)
  47. cv2.imshow('Img', Img)
  48. cv2.imwrite('Img.png', Img)#将画上矩形的图形保存到当前目录
  49. while True:
  50. Key = chr(cv2.waitKey(15) & 255)
  51. if Key == 'q':
  52. cv2.destroyAllWindows()
  53. break

原始图像

处理之后保存的图像

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家自动化/article/detail/154566
推荐阅读
相关标签
  

闽ICP备14008679号