函数:cv2.cvtColor(),cv2.inRange()
转换颜色空间
cv2.cvtColor(input_image,flag),其中 flag就是转换类型。
opencv中常用的颜色转换一般就两种:BGR↔Gray 和BGR↔HSV。对应的flag分别为:cv2.COLOR_BGR2GRAY和cv2.COLOR_BGR2HSV。
PS:HSV 格式中,H(色彩/色度)的取值范围是 [0,179], S(饱和度)的取值范围 [0,255],V(亮度)的取值范围 [0,255]。HSV简介
物体跟踪
在 HSV 颜色空间中要比在 BGR 空间中更容易表示一个特定颜色。我们可以利用这一点来提取带有某个特定颜色的物体。
利用inrange函数过滤视频中的颜色,实现跟踪某一颜色。
代码:
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('frame',frame)
cv2.imshow('mask',mask)
cv2.imshow('res',res)
k=cv2.waitKey(5)&0xFF
if k==27:
break
# 关闭窗口
cv2.destroyAllWindows()
效果:
可以通过下表对应颜色的数值过滤其他颜色
HSV颜色对应RGB的分量范围:
参考:
https://blog.csdn.net/u011321546/article/details/79533669