OpenCV常用基本处理函数（3）颜色空间

颜色空间转换

对图像进行颜色空间转换，比如从 BGR 到灰度图，或者从BGR 到 HSV 等

我们要用到的函数是：cv2.cvtColor(input_image ，flag)，其中 flag就是转换类型，常用的就几种转换的类型；

对于 BGR↔Gray 的转换，我们要使用的 flag 就是 cv2.COLOR_BGR2GRAY。
同样对于 BGR↔HSV 的转换，我们用的 flag 就是 cv2.COLOR_BGR2HSV。

物体跟踪

通过某个颜色来跟踪一个物体，

• 从视频中获取每一帧图像
　　• 将图像转换到 HSV 空间
　　• 设置 HSV 阈值到蓝色范围。
　　• 获取蓝色物体

import cv2
import numpy as np

cap = cv2.VideoCapture(r'你的视频文件')

while(1):

    # Take each frame
    _, frame = cap.read()

    # Convert BGR to HSV
    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)

    # define range of blue color in HSV
    lower_blue = np.array([110,50,50])
    upper_blue = np.array([130,255,255])

    # Threshold the HSV image to get only blue colors
    mask = cv2.inRange(hsv, lower_blue, upper_blue)

    # Bitwise-AND mask and original image
    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()

利用掩膜（mask）进行“与”操作，即掩膜图像白色区域是对需要处理图像像素的保留