基於opencv實現簡單畫板功能
阿新 • • 發佈:2020-08-04
OpenCV提供了很多簡單的語句,實現複雜的功能,根據顏色和滑鼠互動的基礎語句,我們可以建立一個簡單的畫板。儘管它簡單,但是製作的框架步驟不能少。
如果您感興趣,可以按照我的步驟寫一下,或者直接複製總程式中的程式碼執行。
一、功能
明確畫板功能,便於後續編寫程式。
目標功能:可調節顏色,有畫圖區域,可以進行滑鼠輸入。
二、框架搭建
主要有兩步:
第一步:建立顏色選區,可以用OpenCV的滑動條語句cv.createTrackbar設定。cv.createTrackbar一共擁有五個引數設定,它們是(Trackbar名稱,目標視窗名稱,起始值-同時也是預設值,最大值,回撥函式)。
#建立畫布 img=np.zeros((300,512,3),np.uint8) #給畫布命名——需要提供給Trackbar cv.namedWindow("image") #建立RGB三種顏色的Trackbar,返回函式不需要做任何動作 cv.createTrackbar("R","image",255,nothing) cv.createTrackbar("G",nothing) cv.createTrackbar("B",nothing) #由於OpenCv裡面無按鈕函式,所以用0/1來表示開關 switch="0:OFF\n1:ON" cv.createTrackbar(switch,1,nothing) #回撥函式,不需要做任何操作 def nothing(x): pass
第二步:捕捉滑鼠動作狀態,根據滑鼠動作,實現在畫圖區域畫出圖形。常見的滑鼠操作語句,可以使用語句*events = [i for i in dir(cv) if ‘EVENT' in i]*在python內檢視。
#設定一個是否開始畫畫的開關 drawing=False #定義畫畫函式 def draw(event,x,y,flag,param): #將畫畫開關作為全域性變數,以便於在每次動作以後改變它的值 global drawing #使用函式cv.EVENT_LBUTTONDOWN,如果滑鼠左鍵按下,畫圓,同時drawing為真 if event==cv.EVENT_LBUTTONDOWN: cv.circle(img,(x,y),3,(g,b,r),-1) drawing=True #如果滑鼠移動,進行下面操作 elif event==cv.EVENT_MOUSEMOVE: #如果drawing為真,開始畫圓 if drawing==True: cv.circle(img,-1) #如果滑鼠左鍵抬起,畫最後一個圓,並將drawing設為False,此時就會停止畫畫 elif event==cv.EVENT_LBUTTONUP: cv.circle(img,-1) drawing=False
三、主函式
主函式就是對上述兩部分進行整合。
while(1): #展示畫布 cv.imshow("image",img) #設定終止按鈕,為了保證在32/64位系統或者在不是ASCII編碼的鍵盤上有正常的操作,使用&0xFF取低八位 k=cv.waitKey(1)&0xFF #如果k為27,即鍵盤上Esc的ASCII編碼序號,退出畫板 if k==27: break #獲取每個Trackbar中的值 r=cv.getTrackbarPos("R","image") g = cv.getTrackbarPos("G","image") b = cv.getTrackbarPos("B","image") s = cv.getTrackbarPos(switch,"image") #如果開關是0,清空畫板並禁止畫畫,如果是1,則允許 if s==0: img[:]=0 else: cv.setMouseCallback("image",draw) #最後不要忘記銷燬視窗 cv.destroyAllWindows()
四、執行效果
五、總結
從這個簡單的畫板,我們主要學習運用了滑鼠函式cv.setMouseCallback,還有cv.createTrackbar這兩個基本的OpenCV函式,並探索了其中的細節,找到了其中需要注意的地方。
六、總程式
import cv2 as cv import numpy as np def nothing(x): pass img=np.zeros((300,np.uint8) cv.namedWindow("image") cv.createTrackbar("R",nothing) switch="0:OFF\n1:ON" cv.createTrackbar(switch,nothing) drawing=False def draw(event,param): global drawing if event==cv.EVENT_LBUTTONDOWN: cv.circle(img,-1) drawing=True elif event==cv.EVENT_MOUSEMOVE: if drawing==True: cv.circle(img,-1) elif event==cv.EVENT_LBUTTONUP: cv.circle(img,-1) drawing=False while(1): cv.imshow("image",img) k=cv.waitKey(1)&0xFF if k==27: break r=cv.getTrackbarPos("R","image") if s==0: img[:]=0 else: cv.setMouseCallback("image",draw) cv.destroyAllWindows()
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。