Python-OpenCV(2)

GavinZhou

发布于 2018-01-02 15:20:29

9220

发布于 2018-01-02 15:20:29

这次咱们写个有点意思的东西，上个博客在最后写了画线、画矩形之类的，涉及到取颜色(r g b值)，这次咱们就写个图形化的调色板。具体就是：三个滑动条分别代表R、G、B的值，下方有个矩形区域，显示颜色

cv2.createTrackbar(trackbarName, windowName, value, count, onChange) # 创建滑动条 cv2.getTrackbarPos(trackbarname, winname) # 获取滑动条的值

其中的参数含义相信大家看名字就能明白了，不多说了

代码:

__author__ = 'gavinzhou'
# -*- coding: utf-8 -*-

import cv2
import numpy as np


def recall_nothing():
    """
    回调函数，我们不做任何事情所以直接pass
    :return:
    """
    pass


# 创建一幅三通道的图
image = np.zeros((300, 500, 3), dtype=np.uint8)
cv2.namedWindow('Image')

# 创建滑动条
cv2.createTrackbar('R', 'Image', 0, 255, recall_nothing)
cv2.createTrackbar('G', 'Image', 0, 255, recall_nothing)
cv2.createTrackbar('B', 'Image', 0, 255, recall_nothing)

# 创建开关(0表示关闭RGB的调色效果，此时效果为黑)
switch = '0:OFF\n1:ON'
cv2.createTrackbar(switch, 'Image', 0, 1, recall_nothing)


while 1:
    cv2.imshow('Image', image)
    key = cv2.waitKey(1) & 0xFF
    if key == 27:
        # 'ESC键代表退出'
        break
    # 获得各个滑动条的值
    r = cv2.getTrackbarPos('R', 'Image')
    g = cv2.getTrackbarPos('G', 'Image')
    b = cv2.getTrackbarPos('B', 'Image')
    s = cv2.getTrackbarPos(switch, 'Image')

    # 显示
    if s == 0:
        # 开关关闭，显示全黑
        image[:] = 0
    else:
        # 对于彩色图像(RGB),OpenCV按照BGR来显示，Matplotlib按照RGB来显示
        image[:] = [b, g, r]

结果是：