初识OpenCV-Python - 004: Trackbar as the color palette

此次学习了如何用OpenCV建立一个色调盘。其中会用到cv2.getTrackbarPos(), cv2.createTrackbar()函数。

code:

import cv2
import numpy as np

def nothing(x):
 pass

#Create a black image, a window
img = np.zeros((300,512,3),np.uint8)
cv2.namedWindow('image')

#create trackbars for color change
cv2.createTrackbar('R','image',0,255,nothing) #建立一个调节红色的横条
cv2.createTrackbar('G','image',0,255,nothing)
cv2.createTrackbar('B','image',0,255,nothing)

#create switch for ON/OFF functionality
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 == 27:
 break

 #get current positions of four trackbars
 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()

Code结果展示: