我们可以利用BGR
转换到
HSV来提取带有某个特定颜色的物体。在 HSV
颜色空间中要比在
BGR
空间 中更容易表示一个特定颜色。在我们的程序中,我们要提取的是一个蓝色的物 体。下面就是就是我们要做的几步:
1.从视频中获取每一帧图像
将图像转换到 HSV
空间
2.设置
HSV
阈值到蓝色范围。
3.获取蓝色物体,当然我们还可以做其他任何我们想做的事,比如:在蓝色 物体周围画一个圈。
下面就是我们的代码:
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
值
函数
cv2.cvtColor() ,传入的参数是BGR
值。例如,我们要找到绿色的
HSV
值,我们只需在终 端输入以下命令:
import cv2
import numpy as np
green=np.uint8([[[0,255,0]]])
hsv_green=cv2.cvtColor(green,cv2.COLOR_BGR2HSV)
print(hsv_green)
[[[
60 255 255
]]]
现在你可以分别用
[H-100
,
100
,
100]
和
[H+100
,
255
,
255]
做上下阈值。