把鼠標作爲畫筆
目標
- 學會在OpenCv中處理鼠標事件
- 你將要學習這個函數:cv2.setMouseCallback()
4.1 簡單的演示
在這裏,我們創建一個簡單的應用程序,只要雙擊它,它就在圖像上繪製一個圓。
首先,我們創建一個鼠標回調函數,該函數在鼠標事件發生時執行。鼠標事件可以是鼠標上的任何動作,比如左鍵按下,左鍵鬆開,左鍵雙擊等。我們可以通過鼠標事件獲得與鼠標光標對應的圖片上的座標。根據這些信息我們可以做任何我們想做的事。你可以通過執行下列代碼查看所有被支持的鼠標事件。
import cv2
events = [i for i in dir(cv2) if 'EVENT' in i]
print(events)
創建鼠標回調函數有一個特定的形式,在任何地方都是相同的。它們僅僅是函數的功能不同。所以我們的鼠標回調函數只能做一件事,它在鼠標雙擊的地方畫一個圓。下面是代碼,不懂得地方可以看註釋。
import cv2
import numpy as np
#鼠標回調函數
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()
4.2 高級一點的示例
現在我們創建一個更好的程序。這次我們的程序要完成的任務是根據我們選擇的模式在拖動鼠標時繪製矩形或者是圓圈(就像畫圖程序中一樣)。所以我們的回調函數包括兩部分,一部分畫矩形,一部分畫圓。這是一個典型的例子,它可以幫助我們更好的理解與構建人機交互程序,比如物體跟蹤,圖像分割等。
import cv2
import numpy as np
drawing = False #如果按下鼠標,則爲true
mode = True #如果是 True 則畫矩形。按 m 鍵變成繪製曲線。
#鼠標回調函數
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)
下面我們要把這個回調函數與 OpenCv 窗口綁定在一起。在主循環中我們需要講鍵盤上的 “m“ 鍵與模式轉換綁定在一起。
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()