黑马程序员技术交流社区

标题: 【上海校区】原 python opencv入门鼠标绘图（4） [打印本页]

作者: wuqiong 时间: 2018-7-20 10:00
标题: 【上海校区】原 python opencv入门鼠标绘图（4）
内容来自OpenCV-Python Tutorials 自己翻译整理

目标：

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

作者: 不二晨 时间: 2018-7-20 10:52
优秀，奈斯

作者: wuqiong 时间: 2018-7-23 11:35

作者: 摩西摩西OvO 时间: 2018-7-23 13:43

作者: 吴琼老师 时间: 2018-7-26 16:06

欢迎光临黑马程序员技术交流社区 (http://bbs.itheima.com/)

黑马程序员IT技术论坛 X3.2