目标:
学习如何操作鼠标事件
学习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()