python opencv入門 滑鼠繪圖(4)
阿新 • • 發佈:2019-01-03
目標:
- 學習如何操作滑鼠事件
- 學習cv2.setMouseCallback()函式
簡單樣例:
首先建立一個滑鼠的回撥函式,當滑鼠事件觸發時,該函式執行。
滑鼠事件有很多種,比如左鍵右鍵,雙擊等等,該函式提供滑鼠點選的座標。對應事件的觸發和滑鼠點選的座標位置,我們可以做任何事,下面程式碼輸出了滑鼠事件的種類
import numpy as np
import cv2
events = [i for i in dir(cv2) if 'EVENT' in i]
print(events)
輸出結果為
['EVENT_FLAG_ALTKEY', 'EVENT_FLAG_CTRLKEY', 'EVENT_FLAG_LBUTTON', 'EVENT_FLAG_MBUTTON', 'EVENT_FLAG_RBUTTON', 'EVENT_FLAG_SHIFTKEY', 'EVENT_LBUTTONDBLCLK', 'EVENT_LBUTTONDOWN', 'EVENT_LBUTTONUP', 'EVENT_MBUTTONDBLCLK', 'EVENT_MBUTTONDOWN', 'EVENT_MBUTTONUP', 'EVENT_MOUSEHWHEEL', 'EVENT_MOUSEMOVE', 'EVENT_MOUSEWHEEL', 'EVENT_RBUTTONDBLCLK', 'EVENT_RBUTTONDOWN', 'EVENT_RBUTTONUP']
其中的含義為
CV_EVENT_MOUSEMOVE 0 滑動 CV_EVENT_LBUTTONDOWN 1 左鍵點選 CV_EVENT_RBUTTONDOWN 2 右鍵點選 CV_EVENT_MBUTTONDOWN 3 中間點選 CV_EVENT_LBUTTONUP 4 左鍵釋放 CV_EVENT_RBUTTONUP 5 右鍵釋放 CV_EVENT_MBUTTONUP 6 中間釋放 CV_EVENT_LBUTTONDBLCLK 7 左鍵雙擊 CV_EVENT_RBUTTONDBLCLK 8 右鍵雙擊 CV_EVENT_MBUTTONDBLCLK 9 中間釋放
建立一個滑鼠的回撥函式有一個相同的指定格式,唯一不同的地方在於函式的功能,現在編寫一個雙擊滑鼠就畫一個圈的回撥函式
import numpy as np
import cv2
def draw_circle(event,x,y,flags,param):
if event == cv2.EVENT_LBUTTONDBLCLK:
cv2.circle(img,(x,y),100,(255,0,0),-1)
img = np.zeros((512,512,3),np.uint8)
cv2.namedWindow('image')
cv2.setMouseCallback('image' ,draw_circle)
while(1):
cv2.imshow('image',img)
if cv2.waitKey(20) & 0xFF == 27:
break
cv2.destroyAllWindows()
高逼格樣例:
現在實現一個更高階的應用,類似畫圖工具一樣,要麼畫圓,要麼畫矩形,依照選擇的模式來實現,所以滑鼠回撥函式有兩部分,一個用來畫矩形,一個用來畫圓。這個例子灰常有用,尤其是在互動式應用像是目標追蹤和影象分割。
import numpy as np
import cv2
drawing = False #滑鼠按下為真
mode = True #如果為真,畫矩形,按m切換為曲線
ix,iy=-1,-1
def draw_circle(event,x,y,flags,param):
global ix,iy,drawing,mode
if event == cv2.EVENT_LBUTTONDOWN:
drawing = True
ix,iy=x,y
elif event == cv2.EVENT_MOUSEMOVE:
if drawing == True:
if mode == True:
cv2.rectangle(img,(ix,iy),(x,y),(0,255,0),-1)
else:
cv2.circle(img,(x,y),5,(0,0,255),-1)
elif event == cv2.EVENT_LBUTTONUP:
drawing = False
if mode == True:
cv2.rectangle(img,(ix,iy),(x,y),(0,255,0),-1)
else:
cv2.circle(img,(x,y),5,(0,0,255),-1)
img = np.zeros((512,512,3),np.uint8)
cv2.namedWindow('image')
cv2.setMouseCallback('image',draw_circle)
while(1):
cv2.imshow('image',img)
k = cv2.waitKey(1) & 0xFF
if k == ord('m') :
mode = not mode
elif k == 27:
break
cv2.destroyAllWindows()
如何畫一個線框的拖拽矩形?
如果繼續使用上面的程式碼,那麼在滑鼠拖拽的時候就會繪製矩形。現在要實現一個類似矩形圈定的功能,這裡的思路是在拖拽過程中儲存上一次的拖拽座標,然後畫一個黑色的矩形,就相當於把剛剛畫上的矩形給覆蓋上了,不過要求每次圈定的矩形框都保留,所以有一個bug,就是如果矩形框有相交的部分,那麼拖拽的過程中產生的黑色矩形框就會把之前的矩形給擦掉,就這樣吧,想不出別的方法了!
import numpy as np
import cv2
drawing = False #滑鼠按下為真
mode = True #如果為真,畫矩形,按m切換為曲線
ix,iy=-1,-1
px,py=-1,-1
def draw_circle(event,x,y,flags,param):
global ix,iy,drawing,px,py
if event == cv2.EVENT_LBUTTONDOWN:
drawing = True
ix,iy=x,y
elif event == cv2.EVENT_MOUSEMOVE:
if drawing == True:
cv2.rectangle(img,(ix,iy),(px,py),(0,0,0),0)#將剛剛拖拽的矩形塗黑
cv2.rectangle(img,(ix,iy),(x,y),(0,255,0),0)
px,py=x,y
elif event == cv2.EVENT_LBUTTONUP:
drawing = False
cv2.rectangle(img,(ix,iy),(x,y),(0,255,0),0)
px,py=-1,-1
img = np.zeros((512,512,3),np.uint8)
cv2.namedWindow('image')
cv2.setMouseCallback('image',draw_circle)
while(1):
cv2.imshow('image',img)
k = cv2.waitKey(1) & 0xFF
if k == ord('q') :
break
elif k == 27:
break
cv2.destroyAllWindows()