基于opencv實(shí)現(xiàn)簡單畫板功能

更新時間：2020年08月02日 15:07:44 作者：海底撈淡水魚

這篇文章主要為大家詳細(xì)介紹了基于opencv實(shí)現(xiàn)簡單畫板功能，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下

OpenCV提供了很多簡單的語句，實(shí)現(xiàn)復(fù)雜的功能，根據(jù)顏色和鼠標(biāo)交互的基礎(chǔ)語句，我們可以建立一個簡單的畫板。盡管它簡單，但是制作的框架步驟不能少。

如果您感興趣，可以按照我的步驟寫一下，或者直接復(fù)制總程序中的代碼運(yùn)行。

一、功能

明確畫板功能，便于后續(xù)編寫程序。
目標(biāo)功能：可調(diào)節(jié)顏色，有畫圖區(qū)域，可以進(jìn)行鼠標(biāo)輸入。

二、框架搭建

主要有兩步：

第一步：建立顏色選區(qū)，可以用OpenCV的滑動條語句cv.createTrackbar設(shè)置。cv.createTrackbar一共擁有五個參數(shù)設(shè)置，它們是（Trackbar名稱，目標(biāo)窗口名稱，起始值-同時也是默認(rèn)值，最大值，回調(diào)函數(shù)）。

#創(chuàng)建畫布
img=np.zeros((300,512,3),np.uint8)
#給畫布命名——需要提供給Trackbar
cv.namedWindow("image")
#創(chuàng)建RGB三種顏色的Trackbar，返回函數(shù)不需要做任何動作
cv.createTrackbar("R","image",0,255,nothing)
cv.createTrackbar("G","image",0,255,nothing)
cv.createTrackbar("B","image",0,255,nothing)
#由于OpenCv里面無按鈕函數(shù)，所以用0/1來表示開關(guān)
switch="0:OFF\n1:ON"
cv.createTrackbar(switch,"image",0,1,nothing)
#回調(diào)函數(shù)，不需要做任何操作
def nothing(x):
 pass

第二步：捕捉鼠標(biāo)動作狀態(tài)，根據(jù)鼠標(biāo)動作，實(shí)現(xiàn)在畫圖區(qū)域畫出圖形。常見的鼠標(biāo)操作語句，可以使用語句*events = [i for i in dir(cv) if ‘EVENT' in i]*在python內(nèi)查看。

#設(shè)置一個是否開始畫畫的開關(guān)
drawing=False
#定義畫畫函數(shù)
def draw(event,x,y,flag,param):
 #將畫畫開關(guān)作為全局變量，以便于在每次動作以后改變它的值
 global drawing
 #使用函數(shù)cv.EVENT_LBUTTONDOWN，如果鼠標(biāo)左鍵按下，畫圓，同時drawing為真
 if event==cv.EVENT_LBUTTONDOWN:
  cv.circle(img,(x,y),3,(g,b,r),-1)
  drawing=True
 #如果鼠標(biāo)移動，進(jìn)行下面操作
 elif event==cv.EVENT_MOUSEMOVE:
  #如果drawing為真，開始畫圓
  if drawing==True:
   cv.circle(img, (x, y), 3, (g, b, r), -1)
 #如果鼠標(biāo)左鍵抬起，畫最后一個圓，并將drawing設(shè)為False，此時就會停止畫畫
 elif event==cv.EVENT_LBUTTONUP:
  cv.circle(img,(x,y),3,(g,b,r),-1)
  drawing=False

三、主函數(shù)

主函數(shù)就是對上述兩部分進(jìn)行整合。

while(1):
 #展示畫布
 cv.imshow("image",img)
 #設(shè)置終止按鈕，為了保證在32/64位系統(tǒng)或者在不是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")
 #如果開關(guān)是0，清空畫板并禁止畫畫，如果是1，則允許
 if s==0:
  img[:]=0
 else:
  cv.setMouseCallback("image", draw)
#最后不要忘記銷毀窗口
cv.destroyAllWindows()

四、運(yùn)行效果

五、總結(jié)

從這個簡單的畫板，我們主要學(xué)習(xí)運(yùn)用了鼠標(biāo)函數(shù)cv.setMouseCallback，還有cv.createTrackbar這兩個基本的OpenCV函數(shù)，并探索了其中的細(xì)節(jié)，找到了其中需要注意的地方。

六、總程序

import cv2 as cv
import numpy as np

def nothing(x):
 pass
img=np.zeros((300,512,3),np.uint8)
cv.namedWindow("image")
cv.createTrackbar("R","image",0,255,nothing)
cv.createTrackbar("G","image",0,255,nothing)
cv.createTrackbar("B","image",0,255,nothing)
switch="0:OFF\n1:ON"
cv.createTrackbar(switch,"image",0,1,nothing)

drawing=False
def draw(event,x,y,flag,param):
 global drawing
 if event==cv.EVENT_LBUTTONDOWN:
  cv.circle(img,(x,y),3,(g,b,r),-1)
  drawing=True
 elif event==cv.EVENT_MOUSEMOVE:
  if drawing==True:
   cv.circle(img, (x, y), 3, (g, b, r), -1)
 elif event==cv.EVENT_LBUTTONUP:
  cv.circle(img,(x,y),3,(g,b,r),-1)
  drawing=False

while(1):
 cv.imshow("image",img)
 k=cv.waitKey(1)&0xFF
 if k==27:
  break
 r=cv.getTrackbarPos("R","image")
 g = cv.getTrackbarPos("G", "image")
 b = cv.getTrackbarPos("B", "image")
 s = cv.getTrackbarPos(switch, "image")
 if s==0:
  img[:]=0
 else:
  cv.setMouseCallback("image", draw)
cv.destroyAllWindows()

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: