OpenCV-Python教程:5.鼠标作为画笔

简单例子

我们创建一个简单的应用来，我们在图片上双击，就会在那画一个圆

首先我们创建一个鼠标回调函数，当一个鼠标事件发生时来执行。鼠标事件可以是任何和鼠标左键点下，左键抬起，左键双击等相关的时间。它会给我们每次鼠标事件的坐标（x,y）。通过这些事件和坐标，我们可以做任何我们想做得，要列出所有可用的事件，执行下面的代码：

>>>import cv2
>>>events = [i for i in dir(cv2) if 'EVENT' in i]
>>>print events

创建一个鼠标回调函数有通用的格式。区别只是这个函数是做什么的。所以我们的鼠标回调函数只做一件事，双击的时候画一个圆。

import cv2
import numpy as np

# mouse callback function
def draw_circle(event, x, y, flags, param):
    if event == cv2.EVENT_LBUTTONDBLCLK:
        cv2.circle(img, (x, y), 100, (255, 0, 0), -1)

# Create a black image, a window and bind the function to window
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 cv2
import numpy as np

drawing = False     # true if mouse is pressed
mode=True    # if True, draw rectangle. Press 'm' to toggle to curve
ix, iy = -1, -1

# mouse callback function
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()