08 颜色空间的转化

# cvtColor(img,colorspace):颜色转换的的关键API

# cvtColor(img,colorspace):颜色转换的的关键API
import cv2

def callback(value):
    pass

cv2.namedWindow('color', cv2.WINDOW_NORMAL)
cv2.resizeWindow('color', 640, 480)

# 读取照片，Opencv读进来的图片默认是BGR的色彩空间
img = cv2.imread('fj.jpg')


# 定义颜色空间转换列表
color_space = [
    # 所有颜色空间的转化都是COLOR开头
    cv2.COLOR_BGR2RGBA, cv2.COLOR_BGR2BGRA,
    cv2.COLOR_BGR2GRAY, cv2.COLOR_BGR2HSV,
    cv2.COLOR_BGR2YUV
]

# 设置trackbar
cv2.createTrackbar('trackbar', 'color', 0, 4, callback)
while True:
    #获取trackbar的值
    index = cv2.getTrackbarPos('trackbar', 'color')

    # 进行颜色空间转换
    cvt_img = cv2.cvtColor(img, color_space[index])
    cv2.imshow('color', cvt_img)
    key = cv2.waitKey(10)
    if key == ord('q'):
        break

cv2.destroyAllWindows()