学习目标:
- 学会把滑动条绑定到 OpenCV 的窗口。
- 学习函数:cv2.getTrackbarPos(), cv2.creatTrackbar()等。
简单演示:
通过调节滑动条来设定画板颜色。我们要创建一个窗口来显示显色,还有三个滑动条来设置 B,G,R 的颜色。当我们滑动滚动条是窗口的颜色也会发生相应改变。默认情况下窗口的起始颜色为黑。滑动条的另外一个重要应用就是用作转换按钮。默认情况下 OpenCV 本身不带有按钮函数。所以我们使用滑动条来代替。在我们的程序中,我们要创建一个转换按钮,只有当装换按钮指向 ON 时,滑动条的滑动才有用,否则窗户口都是黑的。
import cv2
import numpy as np
def nothing(x):passimg=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)&0xFFif k == ord('q'):breakr=cv2.getTrackbarPos('R','image')g=cv2.getTrackbarPos('G','image')b=cv2.getTrackbarPos('B','image')s=cv2.getTrackbarPos(switch,'image')if s==0:img[:]=0else: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 npdrawing = False
mode = True
ix,iy = -1,-1
def nothing(x):passdef 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,iyif event==cv2.EVENT_LBUTTONDOWN:drawing = Trueix,iy = x,yelif 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 = Falseimg=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'):breakelif cv2.waitKey(1) &0xFF == ord('m'):mode = not mode
cv2.imwrite('/home/wl/1.jpg',img)
cv2.destroyAllWindows()