-
Notifications
You must be signed in to change notification settings - Fork 0
/
get_image_coords.py
59 lines (45 loc) · 1.79 KB
/
get_image_coords.py
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
import matplotlib.pyplot as plt
import argparse
import cv2
def get_points_from_image(image):
# Create a figure and axis
fig, ax = plt.subplots()
# Display the image
ax.imshow(image)
# Initialize a list to store the selected points
points = []
def onclick(event):
# Check if the mouse click is within the image boundaries
if event.xdata is not None and event.ydata is not None:
# Check the current toolbar mode
if plt.get_current_fig_manager().toolbar.mode == '': # Check for no active mode
# Append the clicked point to the list
points.append([event.xdata, event.ydata])
# Plot a red circle at the clicked point
ax.plot(event.xdata, event.ydata, 'ro')
# Number the point
point_number = len(points)
ax.text(event.xdata, event.ydata, str(point_number), color='red', fontsize=12)
plt.draw()
# Connect the click event listener
cid = fig.canvas.mpl_connect('button_press_event', onclick)
# Show the image and wait for user input
plt.show()
# Return the selected points
return points
parser = argparse.ArgumentParser(description="input a list of point on an image")
parser.add_argument("video_path")
args = parser.parse_args()
cap = cv2.VideoCapture(args.video_path)
ret, image = cap.read()
cap.release()
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
selected_points = get_points_from_image(image)
fig, ax = plt.subplots()
ax.imshow(image)
for i, point in enumerate(selected_points):
ax.plot(point[0], point[1], 'ro')
ax.text(point[0], point[1], str(i+1), color='red', fontsize=12)
plt.savefig("./numbered_points_image.png")
# Print the selected points
print("Selected points:", selected_points)