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()

Opencv笔记（六）——把滑动条当调色板的相关教程结束。