OpenCV-Python官方教程-05-颜色空间转换、获取特定颜色图像、物体追踪

• 这里主要用到两个函数
  cv2.cvtColor(img,flag) 用图像的颜色空间转换,flag有很多种，主要用到有cv2.COLOR_BGR2GRAY、cv2.COLOR_BGR2HSV、cv2.COLOR_BGR2LAB等；cv2.inRange(img,lower_thrshold,upper_threshold) 在 HSV 颜色空间中要比在 BGR 空间中更容易表示一个特定颜色。所以在使用这个函数的时候，我们首先需要将图像转换到HSV颜色空间中。lower_thrshold,upper_threshold分别是要查找颜色的高低阈值。

import cv2
import numpy as np
img = cv2.imread('opencv.jpg')
hsv = cv2.cvtColor(img,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(img,img,mask=mask)
cv2.imshow('mask',mask)
cv2.imshow('res',res)
cv2.waitKey(0)
cv2.destroyAllWindows()

• 怎么找到要检测对象的HSV值呢？一般可以先确定其RGB，然后转为HSV。下面给大家推荐一个方法，直接利用QQ或是微信的截图，当箭头指向某点时就会显示其RBG值，之后再转为HSV即可。注意下面是三个中括号[]

green = np.uint8([[[0,255,0]]])
hsv_green = cv2.cvtColor(green,cv2.COLOR_BGR2HSV)
print (hsv_green)

[[[ 60 255 255]]]

• 物体追踪
  1.从视频中获取每一帧图像
  2.将图像转换到 HSV 空间
  3.设置追踪目标的 HSV 阈值范围。

## 追踪
cap = cv2.VideoCapture('skin-v2.mp4')
while(1):
    #获取每帧
    ret,frame = cap.read()
    #转换到HSV
    hsv = cv2.cvtColor(frame,cv2.COLOR_BGR2HSV)
    #设定阈值
    lower_blue = np.array([20,0,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()