1. 程式人生 > 程式設計 >基於opencv實現簡單畫板功能

基於opencv實現簡單畫板功能

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

四、執行效果

基於opencv實現簡單畫板功能

五、總結

從這個簡單的畫板,我們主要學習運用了滑鼠函式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()

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。