OpenCV中经常用到的颜色空间转换,经常用到的有两种BGR-Gray,跟BGR-HSV。
cv2.cetColor(input_image,flag),
flag就是我们要转换的类型,对于灰度图转换,flag就是cv2.COLOR_BGR2GRAY,对于HSV转换,用到的flag是cv2.COLOR_BGR2HSV。
在opencv的HSV格式中,H(hue:色彩/亮度)的取值范围是[0,179],S(saturation:饱和度)取值范围是[0,255],V(value:亮度)的取值范围是[0,255]。当拿opencv的HSV值与其他软件的HSV值进行对比时,一定要记住归一化。
import cv2
import numpy as np
cap=cv2.VideoCapture(0)
while(1):
#获取每一帧
ret,frame=cap.read()
#转换到HSV
hsv=cv2.cvtColor(frame,cv2.COLOR_BGR2HSV)
#设置蓝色闸值
lower_blue=np.array([110,50,50])
upper_blue=np.array([130,255,255])
#根据闸值构建掩膜
mask=cv2.inRange(hsv,lower_blue,upper_blue)
#对原图像和掩膜进行位运算
res=cv2.bitwise_and(frame,frame,mask=mask)
cv2.imshow('hsv',hsv)
cv2.imshow('mask',mask)
cv2.imshow('res',res)
k=res=cv2.waitKey(5)&0xFF
while k==27:
break
cv2.destroyAllWindows()
如何找到跟踪对象的HSV值
import numpy as np
import cv2
green=np.uint8([[[0,255,0]]])
hsv_green=cv2.cvtColor(green,cv2.COLOR_BGR2HSV)
print(hsv_green)
作业: 尝试同时提取多个的不同颜色,比如同时提取红,蓝,绿三个不同颜色的物体
import numpy as np
import cv2
#读取图片
img=cv2.imread('1.jpg',cv2.IMREAD_COLOR)
#BGR转换HSV
hsv=cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
#设置闸值
lower_blue=np.array([110,100,100])#blue
upper_blue=np.array([130,255,255])
lower_green=np.array([50,100,100])#green
upper_green=np.array([70,255,255])
lower_red=np.array([0,100,100])#red
upper_red=np.array([10,255,255])
#根据闸值构建掩膜
blue_mask=cv2.inRange(hsv,lower_blue,upper_blue)
green_mask=cv2.inRange(hsv,lower_green,upper_green)
red_mask=cv2.inRange(hsv,lower_red,upper_red)
#对原图像和掩膜进行位运算
blue=cv2.bitwise_and(img,img,mask=blue_mask)
green=cv2.bitwise_and(img,img,mask=green_mask)
red=cv2.bitwise_and(img,img,mask=red_mask)
res=red+blue+green
cv2.imshow('res',res)
cv2.imwrite('magic1.jpg',res)
cv2.waitKey(0)
cv2.destroyAllWindows()