| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758 |
- import os
- import json
- import cv2
- import numpy as np
- input_folder = r"G:\python_ws_g\data\test\jsonout" # 你的 JSON + 图片输入文件夹
- output_folder = r"G:\python_ws_g\data\test\arcshow" # 保存带标注图片的文件夹
- # input_folder = r"G:\python_ws_g\data\xiao\out" # 你的 JSON + 图片输入文件夹
- # output_folder = r"G:\python_ws_g\data\xiao\arcshow" # 保存带标注图片的文件夹
- MAX_WIDTH = 1280
- MAX_HEIGHT = 720
- def resize_to_fit(image, max_width=MAX_WIDTH, max_height=MAX_HEIGHT):
- h, w = image.shape[:2]
- scale = min(max_width / w, max_height / h, 1.0) # 只缩小不放大
- new_size = (int(w * scale), int(h * scale))
- resized_image = cv2.resize(image, new_size, interpolation=cv2.INTER_AREA)
- return resized_image
- def visualize_and_save(input_folder, output_folder):
- os.makedirs(output_folder, exist_ok=True)
- files = [f for f in os.listdir(input_folder) if f.endswith('.json')]
- files.sort()
- for json_file in files:
- json_path = os.path.join(input_folder, json_file)
- with open(json_path, 'r', encoding='utf-8') as f:
- data = json.load(f)
- image_path = os.path.join(input_folder, data["imagePath"])
- if not os.path.exists(image_path):
- print(f"图像不存在：{image_path}")
- continue
- image = cv2.imread(image_path)
- for shape in data["shapes"]:
- label = shape["label"]
- points = shape["points"]
- for pt in points:
- x, y = int(pt[0]), int(pt[1])
- cv2.circle(image, (x, y), radius=3, color=(0, 255, 0), thickness=-1)
- x0, y0 = int(points[0][0]), int(points[0][1])
- cv2.putText(image, label, (x0, y0 - 5), cv2.FONT_HERSHEY_SIMPLEX, 0.6, (0, 255, 0), 1)
- resized_image = resize_to_fit(image)
- base_name = os.path.splitext(json_file)[0]
- save_path = os.path.join(output_folder, base_name + ".jpg")
- cv2.imwrite(save_path, resized_image)
- print(f"保存图片: {save_path}")
- if __name__ == "__main__":
- visualize_and_save(input_folder, output_folder)
|
