文章目录
- 一、本期目标
- 二、简单演示
- 三、更高级的示例
一、本期目标
阅读本期内容,你将:
- 学习使用 OpenCV 处理鼠标事件
你将要学习的函数是:
- cv2.setMouseCallback()
二、简单演示
这里我们来创建一个简单的程序,他会在图片上你双击过的位置绘制一个圆圈。
首先我们来创建一个鼠标事件回调函数,但鼠标事件发生是他就会被执行。
鼠标事件可以是鼠标上的任何动作,比如左键按下,左键松开,左键双击等。
我们可以通过鼠标事件获得与鼠标对应的图片上的坐标。根据这些信息我们可以做任何我们想做的事。
你可以通过执行下列代码查看所有被支持的鼠标事件。
import cv2
events = [i for i in dir(cv2) if 'EVENT' in i]
print(events)
所有的鼠标事件回调函数都有一个统一的格式,他们所不同的地方仅仅是被调用后的功能。
我们的鼠标事件回调函数只用做一件事:在双击过的地方绘制一个圆圈。
下面是代码,不懂的地方可以看看注释。
import cv2
import numpy
# 回调函数
def draw_circle(event, x , y, flags, param):
if event == cv2.EVENT_LBUTTONDBLCLK:
cv2.circle(image, (x, y), 50, (255, 255, 255), 3)
# 创建图像与窗口并将窗口与回调函数绑定
image = numpy.zeros((512, 512, 3), numpy.uint8)
cv2.namedWindow('image')
cv2.setMouseCallback('image', draw_circle)
while(1):
cv2.imshow('image', image)
# 按下ESC退出
if cv2.waitKey(20) & 0xFF==27:
break
cv2.destroyAllWindows()
三、更高级的示例
现在我们来创建一个更好的程序。
这次我们的程序要完成的任务是:根据我们选择的模式,在拖动鼠标时绘制矩形或者是圆圈(就像画图程序中一样)。
所以我们的回调函数包含两部分,一部分画矩形,一部分画圆圈。
这是一个典型的例子,它可以帮助我们更好理解与构建人机交互式程序,比如物体跟踪,图像分割等。
import cv2
import numpy
# 当鼠标按下时变为 True
drawing = False
# 如果 mode 为 true 绘制矩形。按下 'm' 变成绘制曲线。
mode = True
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
# 当鼠标左键按下并移动是绘制图形。 event 可以查看移动, flag 查看是否按下
elif event == cv2.EVENT_MOUSEMOVE and flags == cv2.EVENT_FLAG_LBUTTON:
if drawing == True:
if mode == True:
cv2.rectangle(image, (ix, iy), (x, y), (0, 255, 0), -1)
else:
# 绘制圆圈,小圆点连在一起就成了线, 3 代表了笔画的粗细
cv2.circle(image, (x, y), 3, (0, 0, 255), -1)
# 下面注释掉的代码是起始点为圆心,起点到终点为半径的
# r = int(numpy.sqrt((x-ix)**2+(y-iy)**2))
# cv2.circle(image, (x, y), r, (0, 0, 255), -1)
# 当鼠标松开停止绘画。
elif event == cv2.EVENT_LBUTTONUP:
drawing == False
# if mode == True:
# cv2.rectangle(image, (ix, iy), (x, y), (0, 255, 0), -1)
# else:
# cv2.circle(image, (x, y), 5, (0, 0, 255), -1)
image = numpy.zeros((512, 512, 3), numpy.uint8)
cv2.namedWindow('image')
cv2.setMouseCallback('image', draw_circle)
while True:
cv2.imshow('image', image)
k = cv2.waitKey(1) & 0xFF
if k == ord('m'):
mode = not mode
elif k == 27:
break
cv2.destroyAllWindows()
这个示例中,我们绘制的是填充的矩形,尝试修改代码,使其变成不填充的矩形!