xml篇—提取VOC格式的坐標，并按照cameraID進行排序（二）

import os
import xml.etree.ElementTree as ETdef parse_xml(xml_file):tree = ET.parse(xml_file)root = tree.getroot()objects = {}for obj in root.findall('object'):name = obj.find('name').textbbox = obj.find('bndbox')xmin = int(bbox.find('xmin').text)ymin = int(bbox.find('ymin').text)xmax = int(bbox.find('xmax').text)ymax = int(bbox.find('ymax').text)if name not in objects:objects[name] = []objects[name].append((xmin, ymin, xmax, ymax))return objectsdef get_camera_id(filename):# 從文件中提取攝像頭IDreturn filename.split('-')[0]def main():folder_path = '/app/yyq/dataset/profile_materials/camera_point/jiuting_picture/xml'  # 修改為你的XML文件夾路徑xml_files = [f for f in os.listdir(folder_path) if f.endswith('.xml')]# 根據攝像頭ID排序文件列表xml_files_sorted = sorted(xml_files, key=lambda x: get_camera_id(x))cameraid_dict = {}links = []for xml_file in xml_files_sorted:xml_path = os.path.join(folder_path, xml_file)objects = parse_xml(xml_path)camera_id = xml_file.split('-')[0]if "1" in objects:cameraid_dict[camera_id] = objects["1"][0] if objects["1"] else Noneif "2" in objects:links.append({"name": camera_id,"box": objects["2"]})print("CAMERAID_DICT = {")for camera_id, bbox in cameraid_dict.items():print(f'    "{camera_id}": {bbox},')print("}")print("\nLINKS = [")for link in links:print("    {")print(f'        "name": "{link["name"]}",')print(f'        "box": {link["box"]}')print("    },")print("]")if __name__ == '__main__':main()

結果如下：

CAMERAID_DICT = {"3101171070000000": (700, 396, 874, 614),"3101171070000001": (272, 741, 1178, 1080),
}LINKS = [{"name": "3101171070000000","box": [(823, 396, 874, 432)]},{"name": "3101171070000001","box": [(888, 739, 1180, 893), (274, 987, 425, 1076)]},
]