Demo đơn giản

Ở bước này, chúng ta sẽ tạo một ứng dụng đơn giản để vẽ các hình tròn trên hình ảnh ở bất cứ vị trí nào bằng cách nhấp đúp chuột vào đó.

Đầu tiên chúng ta tạo một hàm gọi lại chuột được thực thi khi một sự kiện chuột diễn ra. Sự kiện chuột có thể là bất kỳ thứ gì liên quan đến chuột máy tính như chuột trái, chuột phải, nhấp đúp chuột trái, ... Hàm này cung cấp cho chúng ta tọa độ (x, y) cho mọi sự kiện chuột. Để liệt kê tất cả các sự kiện có sẵn, hãy chạy đoạn code sau đây:

1. import cv2
2. events = [i for i in dir(cv2) if 'EVENT' in i]
3. print events

Kết quả sau khi chạy đoạn code trên:

Tạo chức năng gọi lại chuột có một định dạng cụ thể giống nhau ở mọi nơi. Nó chỉ khác nhau về chức năng. Ở ví dụ này, hàm gọi lại chuột thực hiện một việc, đó là vẽ một vòng tròn ở vị trí nhấp đúp chuột. Chạy đoạn code dưới đây để rõ hơn về điều này:

1. import cv2
2. import numpy as np
4. # mouse callback function
5. def draw_circle(event, x, y, flags, param):
6.      if event == cv2.EVENT_LBUTTONDBLCLK:
7.           cv2.circle(img, (x,y), 100, (255,0,0), -1)
9. # Create a black image, a window and bind the function to window
10. img = np.zeros((512, 512, 3), np.uint8)
11. cv2.namedWindow('image')
12. cv2.setMouseCallback('image', draw_circle)
14. while(1):
15.      cv2.imshow('image',img)
16.      if cv2.waitKey(20) & 0xFF == 27:  //Nhan phim “esc” de thoat
17.           break
18. cv2.destroyAllWindows()

Kết quả sau khi chạy đoạn code trên:

Demo nâng cao

Bây giờ chúng ta sẽ phát triển lên một ứng dụng phức tạp hơn. Trong phần này, chúng ta vẽ hình chữ nhật hoặc hình tròn (tùy thuộc vào chế độ chọn) bằng cách kéo chuột giống như được thực hiện trong ứng dụng Paint. Vì vậy, hàm gọi lại chuột lúc này có hai phần: một để vẽ hình chữ nhật và một để vẽ các hình tròn. Ví dụ cụ thể này sẽ thực sự hữu ích trong việc tạo và hiểu một số ứng dụng tương tác như theo dõi đối tượng, phân đoạn hình ảnh, ... Xem và chạy đoạn code dưới đây:

1. import cv2
2. import numpy as np
4. drawing = False # true if mouse is pressed
5. mode = True # if True, draw rectangle. Press 'm' to toggle to curve
6. ix, iy = -1, -1
8. # mouse callback function
9. def draw_circle(event, x, y, flags, param):
10.      global ix, iy, drawing, mode
12.      if event == cv2.EVENT_LBUTTONDOWN:
13.           drawing = True
14.           ix, iy = x, y
16.      elif event == cv2.EVENT_MOUSEMOVE:
17.           if drawing == True:
18.                if mode == True:
19.                     cv2.rectangle(img, (ix,iy), (x,y), (0,255,0), -1)
20.                else:
21.                     cv2.circle(img, (x,y), 5, (0,0,255), -1)
23.      elif event == cv2.EVENT_LBUTTONUP:
24.           drawing = False
25.           if mode == True:
26.                cv2.rectangle(img, (ix, iy), (x, y), (0, 255, 0), -1)
27.           else:
28.                cv2.circle(img, (x, y), 5, (0, 0, 255), -1)

Tiếp theo, chúng ta phải ràng buộc chức năng gọi lại chuột này với cửa sổ OpenCV. Trong vòng lặp chính, chúng ta nên đặt liên kết bàn phím cho phím 'm' để chuyển đổi giữa hình chữ nhật và hình tròn.

29. img = np.zeros((512, 512, 3), np.uint8)
30. namedWindow('image')
31. setMouseCallback('image', draw_circle)
33. while(1):
34.      imshow('image',img)
35.      k = cv2.waitKey(1) & 0xFF
36.      if k == ord('m'):
37.           mode = not mode
38.      elif k == 27:
39.           break
41. destroyAllWindows()

Kết quả sau khi chạy đoạn code trên:

Ở phần tiếp theo chúng ta sẽ tìm hiểu cách hiển thị bảng chỉnh màu sắc thông qua các hàm của OpenCV.

(Sưu tầm)
VIỆN IMC
Tòa nhà IMC Tower, Số 176 Trường Chinh, Phường Khương
Thượng, Quận Đống Đa, Thành phố Hà Nội, Việt Nam
Tel/Fax : (+84) 24 3566 6232 / 24 3566 6234
Email: contact@imc.org.vn   Website: https://imc.org.vn