CVtools_ForNUEDC/GetROI.py at master · CodeCatMeow/CVtools_ForNUEDC · GitHub

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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
"""
工具：使用鼠标拖动确定ROI区域，命令行输出坐标和归一化坐标
"""

import cv2

from lib import Video


WIDTH = 640
HEIGHT = 480
point = [[0, 0], [0, 0]]
frame = cv2.VideoCapture()


def getMouse(event, x, y, flags, param):
    global HEIGHT, WIDTH, point
    if event == cv2.EVENT_LBUTTONDOWN:  # 鼠标左键单击
        point[0][0], point[0][1] = x, y
        # print(point[0])
    elif event == cv2.EVENT_MOUSEMOVE and (
        flags & cv2.EVENT_FLAG_LBUTTON
    ):  # 鼠标左键按下拖动
        point[1][0], point[1][1] = x, y
        # print(point[1])
    elif event == cv2.EVENT_LBUTTONUP:
        for p in point:
            if p[0] > WIDTH:
                p[0] = WIDTH
            if p[1] > HEIGHT:
                p[1] = HEIGHT
        print(
            "ROI:  (x1, y1) = (%d, %d), (x2, y2) = (%d, %d)"
            % (point[0][0], point[0][1], point[1][0], point[1][1])
        )
        print(
            "Normalize:  (x1, y1) = (%.3f, %.3f), (x2, y2) = (%.3f, %.3f)"
            % (
                point[0][0] / WIDTH,
                point[0][1] / HEIGHT,
                point[1][0] / WIDTH,
                point[1][1] / HEIGHT,
            )
        )
        print()


cap = Video.CaptureInit(0, PI_MODE=False)

cv2.namedWindow("video", cv2.WINDOW_AUTOSIZE)
cv2.setMouseCallback("video", getMouse)
while True:
    _, frame = cap.read()

    cv2.rectangle(frame, tuple(point[0]), tuple(point[1]), (255, 0, 0), 2)
    cv2.imshow("video", frame)

    content = cv2.waitKey(1) & 0xFF
    if content == 27:  # Esc
        print("exit!")
        break

cv2.destroyAllWindows()