Python实现利用仅有像素级标注的json文件生成框标注的json文件，并存放到新文件夹

import json
import os
# create rectangle labels based on polygon labels, and store in a new folder
def create_rectangle_shapes(polygon_shapes):rectangle_shapes = []for polygon_shape in polygon_shapes:# 获取多边形的坐标点points = polygon_shape['points']# 找到最左上角和最右下角的坐标left = min(points, key=lambda x: x[0])[0]bottom = min(points, key=lambda x: x[1])[1]right = max(points, key=lambda x: x[0])[0]top = max(points, key=lambda x: x[1])[1]# 创建新的矩形形状rectangle_shape = {"label": polygon_shape['label'],"points": [[left, bottom],[right, top],],"group_id": polygon_shape['group_id'],"shape_type": "rectangle","flags": polygon_shape['flags']}rectangle_shapes.append(rectangle_shape)return rectangle_shapesdef main(json_file, target_file):# 读取原始JSON文件with open(json_file, 'r', encoding='utf-8') as f:data = json.load(f)# 删除所有shape_type为rectangle的标注data['shapes'] = [shape for shape in data['shapes'] if shape['shape_type'] != 'rectangle']# 生成新的shape_type为rectangle的标注rectangle_shapes = create_rectangle_shapes(data['shapes'])# 将新的矩形形状添加到原始数据中data['shapes'].extend(rectangle_shapes)# 创建输出文件夹# output_dir = r'D:\dataset\dataset\1_seg_rect'# if not os.path.exists(output_dir):#     os.makedirs(output_dir)# 保存新的JSON文件# with open(os.path.join(output_dir, r'20231016-105404-812_9.json'), 'w') as f:with open(target_file, 'w') as f:json.dump(data, f, indent=4)# 运行脚本
if __name__ == "__main__":jsonF = []newF = []for i in range(1,7):  # 此处替换文件目录地址json_folder = r'D:\dataset\dataset'  # 存放json文件的目录new_folder = r"D:\dataset\dataset"  # 存放新json文件的目录json_folder += '\\' + str(i) + '_seg_without_rect'new_folder += '\\' + str(i) + '_seg_rect'if os.path.exists(json_folder):jsonF.append(json_folder)newF.append(new_folder)if not os.path.exists(new_folder):os.makedirs(new_folder)else:print("not found file")for i in range(len(jsonF)):for filename in os.listdir(jsonF[i]):if filename.split(".")[-1]=="json":f = os.path.join(jsonF[i], filename)o = os.path.join(newF[i], filename)main(f,o)print(f+" completed")