python opencv入门 鼠标绘图（4）

内容来自OpenCV-Python Tutorials 自己翻译整理

目标：

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()