本篇博客介绍将图片从RGB颜色空间转换到HSV颜色空间,从而实现对特定颜色的对象的识别。
下面是代码(实现对浅蓝色物体的识别):
import cv2 import numpy as np rgb_img = cv2.imread('predict1.png') HSV = cv2.cvtColor(rgb_img, cv2.COLOR_BGR2HSV) H, S, V = cv2.split(HSV) lowerBlue = np.array([100, 100, 50]) upperBlue = np.array([130, 255, 255]) mask = cv2.inRange(HSV, lowerBlue, upperBlue) blueThings = cv2.bitwise_and(rgb_img, rgb_img, mask=mask) cv2.imshow('resource', rgb_img) cv2.imshow('result', blueThings) cv2.waitKey(0) cv2.destroyAllWindows()
测试图片:
识别结果: