Mouse로 그리기

Goal

  • Mouse Event의 적용 방법에 대해서 알 수 있다.
  • cv2.setMouseCallback() 함수에 대해서 알 수 있다.

작동방법

OpenCV에는 이미 Mouse Event의 종류에 대해서 사전 정의가 되어 있습니다. 확인을 하기 위해서 Python Terminal에서 아래와 같이 입력해보시기 바랍니다.

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

실행을 하면 다양한 Mouse Event의 종류를 알 수 있습니다. 어떤 종류의 Event인지는 이름을 보면 쉽게 알 수 있습니다.:

'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'

Mouse Event를 확인하고 Callback을 호출하는 함수가 cv2.setMouseCallback() 입니다.

cv2.setMouseCallback(windowName, callback, param=None)
Parameters:
  • windowName – windowName
  • callback – callback함수. callback함수에는 (event, x, y, flags, param)가 전달 됨.
  • param – callback함수에 전달되는 Data

간단한 Demo

아래 Demo는 화면에 Double-Click을 하면 원이 그려지는 예제입니다.

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
import cv2
import numpy as np

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

# 빈 Image 생성
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(0) & 0xFF == 27:
        break

cv2.destroyAllWindows()

Advanced Demo

다음은 마우스를 누른 상태에서 이동시 원 또는 사각형을 그리는 Demo입니다. 이 예제는 향후 대상 추적이나 이미지 Segmentaion시 응용될 수 있습니다.(ex; 이미지에서 대상을 마우스로 선택하고 동일한 대상을 찾는 경우)

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
#-*- coding:utf-8 -*-
import cv2
import numpy as np 

drawing = False #Mouse가 클릭된 상태 확인용
mode = True # True이면 사각형, false면 원
ix,iy = -1,-1


# Mouse Callback함수
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),(255,0,0),-1)
			else:
				cv2.circle(img,(x,y),5,(0,255,0),-1)

	elif event == cv2.EVENT_LBUTTONUP:
		drawing = False;             # 마우스를 때면 상태 변경
		if mode == True:
			cv2.rectangle(img,(ix,iy),(x,y),(255,0,0),-1)
		else:
			cv2.circle(img,(x,y),5,(0,255,0),-1)


img = np.zeros((512,512,3), np.uint8)
cv2.namedWindow('image')
cv2.setMouseCallback('image',draw_circle)

while True:
	cv2.imshow('image', img)

	k = cv2.waitKey(1) & 0xFF

	if k == ord('m'):    # 사각형, 원 Mode변경
		mode = not mode
	elif k == 27:        # esc를 누르면 종료
		break

cv2.destroyAllWindows()