50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
|
import cv2
|
|||
|
|
import numpy as np
|
|||
|
|
import matplotlib.pyplot as plt
|
|||
|
|
|
|||
|
|
# 加载地图图片
|
|||
|
|
image_path = 'map.png' # 替换为你的地图图片路径
|
|||
|
|
image = cv2.imread(image_path)
|
|||
|
|
if image is None:
|
|||
|
|
raise FileNotFoundError("地图图片未找到，请检查路径！")
|
|||
|
|
|
|||
|
|
# 转换为RGB格式，方便使用matplotlib显示
|
|||
|
|
image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
|
|||
|
|
|
|||
|
|
# 创建一个函数，在指定像素坐标画圆
|
|||
|
|
def draw_circle(img, coordinates, color=(255, 0, 0), radius=10, thickness=2):
|
|||
|
|
"""在给定图片上画圆"""
|
|||
|
|
x, y = coordinates
|
|||
|
|
cv2.circle(img, (x, y), radius, color, thickness)
|
|||
|
|
return img
|
|||
|
|
|
|||
|
|
# 复制地图以进行标注
|
|||
|
|
annotated_image = image_rgb.copy()
|
|||
|
|
|
|||
|
|
while True:
|
|||
|
|
try:
|
|||
|
|
# 用户输入像素坐标
|
|||
|
|
user_input = input("请输入像素坐标（格式：x,y）或输入'q'退出：")
|
|||
|
|
if user_input.lower() == 'q':
|
|||
|
|
print("退出程序。")
|
|||
|
|
break
|
|||
|
|
|
|||
|
|
# 解析输入坐标
|
|||
|
|
x, y = map(int, user_input.split(","))
|
|||
|
|
|
|||
|
|
# 检查坐标是否超出范围
|
|||
|
|
if not (0 <= x < image_rgb.shape[1] and 0 <= y < image_rgb.shape[0]):
|
|||
|
|
print("输入的坐标超出图片范围，请重新输入！")
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
# 在指定位置画圆
|
|||
|
|
annotated_image = draw_circle(annotated_image, (x, y))
|
|||
|
|
|
|||
|
|
# 使用matplotlib显示更新后的图片
|
|||
|
|
plt.imshow(annotated_image)
|
|||
|
|
plt.axis('off')
|
|||
|
|
plt.show()
|
|||
|
|
|
|||
|
|
except ValueError:
|
|||
|
|
print("输入格式错误，请输入像素坐标，格式为x,y，例如100,200！")
|