python opencv：使用滑动条做调色板

Posted 2021-03-14 Aloys Wang

cv2.getTrackbarPos() 函数的
一个参数是滑动条的名字，
第二个参数是滑动条被放置窗口的名字，
第三个参数是滑动条的默认位置。
第四个参数是滑动条的最大值，
第五个函数是回调函数，每次滑动条的滑动都会调用回调函数。
回调函数通常都会含有一个默认参数，就是滑动条的位置。
在本例中这个函数不用做任何事情，我们只需要 pass 就可以了
滑动条的另外一个重要应用就是用作转换按钮。
默认情况下 OpenCV 本身不带有按钮函数。
所以我们使用滑动条来代替。
在我们的程序中，我们要创建一个转换按钮，只有当装换按钮指向 ON 时，滑动条的滑动才有用，否则窗户口都是黑的

import cv2
import numpy as np


def nothing(x):
    pass


# 创建一副黑色图像
img = np.zeros((300, 512, 3), np.uint8)
cv2.namedWindow('image')
cv2.createTrackbar('R', 'image', 0, 255, nothing)
cv2.createTrackbar('G', 'image', 0, 255, nothing)
cv2.createTrackbar('B', 'image', 0, 255, nothing)
switch = '0:OFF
1:ON'
cv2.createTrackbar(switch, 'image', 0, 1, nothing)
while(1):
    cv2.imshow('image', img)
    k = cv2.waitKey(1) & 0xFF
    if k == 27:
        break
    r = cv2.getTrackbarPos('R', 'image')
    g = cv2.getTrackbarPos('G', 'image')
    b = cv2.getTrackbarPos('B', 'image')
    s = cv2.getTrackbarPos(switch, 'image')
    if s == 0:
        img[:] = 0
    else:
        img[:] = [b, g, r]
cv2.destroyAllWindows()

import cv2
import numpy as np


def nothing(x):
    pass


# 当鼠标按下时变为 True
drawing = False
# 如果 mode 为 true 绘制矩形。按下'm' 变成绘制曲线。
mode = True
ix, iy = -1, -1
# 创建回调函数


def draw_circle(event, x, y, flags, param):
    r = cv2.getTrackbarPos('R', 'image')
    g = cv2.getTrackbarPos('G', 'image')
    b = cv2.getTrackbarPos('B', 'image')
    color = (b, g, r)
    global ix, iy, drawing, mode
# 当按下左键是返回起始位置坐标
    if event == cv2.EVENT_LBUTTONDOWN:
        drawing = True
        ix, iy = x, y
# 当鼠标左键按下并移动是绘制图形。 event 可以查看移动， flag 查看是否按下
    elif event == cv2.EVENT_MOUSEMOVE and flags == cv2.EVENT_FLAG_LBUTTON:
        if drawing == True:
            if mode == True:
                cv2.rectangle(img, (ix, iy), (x, y), color, -1)
            else:
                # 绘制圆圈，小圆点连在一起就成了线， 3 代表了笔画的粗细
                cv2.circle(img, (x, y), 3, color, -1)
        # 下面注释掉的代码是起始点为圆心，起点到终点为半径的
        # r=int(np.sqrt((x-ix)**2+(y-iy)**2))
        # cv2.circle(img,(x,y),r,(0,0,255),-1)
        # 当鼠标松开停止绘画。
    elif event == cv2.EVENT_LBUTTONUP:
        drawing == False


        # if mode==True:
        # cv2.rectangle(img,(ix,iy),(x,y),(0,255,0),-1)
        # else:
        # cv2.circle(img,(x,y),5,(0,0,255),-1)
img = np.zeros((512, 512, 3), np.uint8)
cv2.namedWindow('image')
cv2.createTrackbar('R', 'image', 0, 255, nothing)
cv2.createTrackbar('G', 'image', 0, 255, nothing)
cv2.createTrackbar('B', 'image', 0, 255, nothing)
cv2.setMouseCallback('image', draw_circle)
while(1):
    cv2.imshow('image', img)
    k = cv2.waitKey(1) & 0xFF
    if k == ord('m'):
        mode = not mode
    elif k == 27:
        break