Opencv3 tutorial_3 色彩空间的转换

色彩空间的转换

读取视频并且追踪视频中标记的某种色彩
通过变化彩色空间，可以实现对视频里的某一特有颜色进行追踪
先拆分通道，修改图像通道颜色，再混合三种颜色

import cv2 as cv
import numpy as np


def extrace_object_demo():
    capture = cv.VideoCapture("C:/Users/LENOVO/Desktop/test image/RPReplay_Final.MP4")
    while(True):
        ret, frame = capture.read()  #  打开摄像头
        if ret == False:
            break
        hsv = cv.cvtColor(frame, cv.COLOR_BGR2HSV)   #  转换为hsv图像  # 通过变化彩色空间，可以实现对视频里的某一特有颜色进行追踪
        lower_hsv = np.array([35, 43, 46])        #  下阈值  # 追踪物体h s v颜色值分布，lower和upper
        upper_hsv = np.array([77, 255, 255])      #  上阈值
        mask = cv.inRange(hsv, lowerb = lower_hsv, upperb = upper_hsv)  # 对我们要追踪的颜色进行标记
        #  cv.inRange(目标图像，阈值下，阈值上，输出图像)
        dst = cv.bitwise_and(frame, frame, mask = mask)      # 可以实现追踪的标记显示为彩色

        cv.imshow("video", frame)
        cv.imshow("mask", dst)
        # cv.imshow("mask", mask)
        c = cv.waitKey(40)
        if c == 27:   # 相当于人为退出
            break

# 色彩空间HSV中 H: 0-180    s: 0-255    V: 0-255
def color_space_demo(image):    # 色彩空间相互转换，一幅彩色图画可以转换成其他色彩空间的图像，最常见的是HSV和RGB  YUV和RGB之间的转换
    gray = cv.cvtColor(image, cv.COLOR_BGR2GRAY)
    cv.imshow("gray", gray)
    hsv = cv.cvtColor(image, cv.COLOR_BGR2HSV)
    cv.imshow("hsv", hsv)
    yuv = cv.cvtColor(image, cv.COLOR_BGR2YUV)
    cv.imshow("yuv", yuv)
    Ycrcb = cv.cvtColor(image, cv.COLOR_BGR2YCrCb)
    cv.imshow("Ycrcb", Ycrcb)


print("----------------hi.python!---------------------------")
src = cv.imread("C:/Users/LENOVO/Desktop/test image/ucc.jpg")
cv.namedWindow("input image",cv.WINDOW_AUTOSIZE)  # 创建一个window
cv.imshow("input image",src)  # 通过窗口名字调用读取的图片到当前的窗口上面

[B, G, R] = cv.split(src)  # 通道拆分  #顺序是b,g,r，不是r,g,b
cv.imshow("blue", B)
cv.imshow("green", G)
cv.imshow("red", R)

src = cv.merge([B, G, R] )      # 通道合并 # 对修改过的颜色进行混合
src[ : , : , 0] = 0   # 可以修改任意通道值达到混合效果
cv.imshow("changed image", src)

color_space_demo(src)
extrace_object_demo()

cv.waitKey(0)            # 告诉系统需要delay，等待用户的下一个操作
cv.destroyAllWindows()   # 响应操作以后就会释放内存