import cv2 import numpy as np import PIL.ImageDraw SCALE = 0.65156853729882650681169151675877 # m/px def add_chinese_text(img, text, position, textColor, textSize): img = PIL.Image.fromarray(img) draw = PIL.ImageDraw.Draw(img) fontStyle = PIL.ImageFont.truetype('simsun.ttc', textSize, encoding='utf-8') draw.text(position, text, textColor, font=fontStyle, stroke_width=1) return np.asarray(img) def main(): # 读取verts with open('verts.txt', 'r', encoding='utf8') as f: verts = f.readlines() verts = list(map(lambda x: x.split(), verts)) verts = list(map(lambda x: [x[0], int(x[1]), int(x[2])], verts)) print(f'{verts=}') # 读取edges with open('edges.txt', 'r', encoding='utf8') as f: edges = f.readlines() edges = list(map(lambda x: x.split(), edges)) edges = list(map(lambda x: [int(x[0]), int(x[1])], edges)) print(f'{edges=}') # 显示地图 im = cv2.imread('map.png') for edge in edges: v1 = verts[edge[0]] # 边关联的点1 v2 = verts[edge[1]] # 边关联的点2 weight = int((((v1[1] - v2[1]) ** 2 + (v1[2] - v2[2]) ** 2) ** 0.5) * SCALE) # 权重为两点欧氏距离 cv2.line(im, (v1[1], v1[2]), (v2[1], v2[2]), (255, 0, 0), 5) # 绘制边的直线 im = add_chinese_text(im, str(weight), ((v1[1] + v2[1]) // 2, (v1[2] + v2[2]) // 2), (255, 0, 255), 30) # 绘制边的权重 for i, vert in enumerate(verts): cv2.circle(im, (vert[1], vert[2]), 20, (255, 0, 0), 2) # 绘制顶点圆圈 im = add_chinese_text(im, str(i), (vert[1] - 15, vert[2] - 15), (255, 0, 255), 30) # 绘制顶点圈中序号数字 im = add_chinese_text(im, vert[0], (vert[1] - 40, vert[2] + 25), (255, 0, 255), 30) # 绘制顶点下方文字 cv2.namedWindow('map', cv2.WINDOW_KEEPRATIO) cv2.imshow('map', im) cv2.waitKey() if __name__ == '__main__': main()

这段代码导入了三个模块：

1. cv2：这是一个用于计算机视觉的库，可以用来处理图像和视频数据。

2. numpy：这是一个用于科学计算的库，主要用于处理数组和矩阵的运算。

3. PIL.ImageDraw：这是 Python Imaging Library (PIL) 中的一个模块，用于在图像上绘图。

SCALE 这个常量的值是一个浮点数，代表一个缩放比例。