Opencv笔记（六）——把滑动条当调色板

学习目标：

• 学会把滑动条绑定到 OpenCV 的窗口。
• 学习函数:cv2.getTrackbarPos(), cv2.creatTrackbar()等。

简单演示：

通过调节滑动条来设定画板颜色。我们要创建一个窗口来显示显色,还有三个滑动条来设置 B,G,R 的颜色。当我们滑动滚动条是窗口的颜色也会发生相应改变。默认情况下窗口的起始颜色为黑。滑动条的另外一个重要应用就是用作转换按钮。默认情况下 OpenCV 本身不带有按钮函数。所以我们使用滑动条来代替。在我们的程序中,我们要创建一个转换按钮,只有当装换按钮指向 ON 时,滑动条的滑动才有用,否则窗户口都是黑的。

import cv2
import numpy as np
def nothing(x):
    pass

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)
switch='0:OFF\n1:ON'
cv2.createTrackbar(switch,'image',0,1,nothing)
while(1):
    cv2.imshow('image',img)
    k=cv2.waitKey(1)&0xFF
    if k == ord('q'):
        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.imwrite('/home/wl/1.jpg',img)
cv2.destroyAllWindows()

• cv2.createTrackbar()是Opencv中的API，其可在显示图像的窗口中快速创建一个滑动控件，用于手动调节阈值，具有非常直观的效果。第一个参数是滑动空间的名称；第二个参数是滑动空间用于依附的图像窗口的名称；第三个参数是初始值；第四个参数是滑动条的最大值，即所有滑动条的数据变动都要在0-count之间，滑动条最小值为0；第五个参数是是回调函数。

• cv2.getTrackbarPos()函数的作用是获取滑动条位置的值。第一个参数是滑动条的名字，第二个参数是窗口的名字。

进一步演示：

结合上一篇博客的绘画函数实现绘图板功能。

import cv2
import numpy as np

drawing = False
mode = True
ix,iy = -1,-1
def nothing(x):
    pass

def draw_circle(event,x,y,flags,param):
    r = cv2.getTrackbarPos('R', 'image')
    g = cv2.getTrackbarPos('G', 'image')
    b = cv2.getTrackbarPos('B', 'image')
    s = cv2.getTrackbarPos('eraser', 'image')
    thin = cv2.getTrackbarPos('thin', 'image')
    color = (b,g,r)
    if s==1:
        color = (255,255,255)

    global drawing,mode,ix,iy
    if event==cv2.EVENT_LBUTTONDOWN:
        drawing = True
        ix,iy = x,y
    elif event==cv2.EVENT_MOUSEMOVE :
        if drawing == True:
            if mode == True:
                cv2.rectangle(img,(ix,iy),(x,y),color,-1)
            else:
                cv2.circle(img,(x,y),thin,color,-1)
    elif event == cv2.EVENT_LBUTTONUP:
        drawing = False

img=np.zeros((512,512,3),np.uint8)
img[ : ] = 255
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.createTrackbar('eraser','image',0,1,nothing)
cv2.createTrackbar('thin','image',1,50,nothing)
cv2.setMouseCallback('image',draw_circle)
while(1):
    cv2.imshow('image',img)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break
    elif cv2.waitKey(1) &0xFF == ord('m'):
        mode = not mode
cv2.imwrite('/home/wl/1.jpg',img)
cv2.destroyAllWindows()