1. 程式人生 > >python opencv入門 滑鼠繪圖(4)

python opencv入門 滑鼠繪圖(4)

目標:

  1. 學習如何操作滑鼠事件
  2. 學習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()