1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
| import os
import json
import xml.etree.ElementTree as ET
def convert_json_to_xml(json_data):
root = ET.Element("annotation")
filename = ET.SubElement(root, "filename")
filename.text = json_data["imagePath"]
object_num = ET.SubElement(root, "object_num")
object_num.text = str(len(json_data["shapes"]))
size = ET.SubElement(root, "size")
width = ET.SubElement(size, "width")
width.text = str(json_data["imageWidth"])
height = ET.SubElement(size, "height")
height.text = str(json_data["imageHeight"])
for shape in json_data["shapes"]:
object_element = ET.SubElement(root, "object")
name = ET.SubElement(object_element, "name")
name.text = shape["label"]
difficult = ET.SubElement(object_element, "difficult")
difficult.text = "0"
bndbox = ET.SubElement(object_element, "bndbox")
xmin = ET.SubElement(bndbox, "xmin")
xmin.text = str(int(shape["points"][0][0]))
ymin = ET.SubElement(bndbox, "ymin")
ymin.text = str(int(shape["points"][0][1]))
xmax = ET.SubElement(bndbox, "xmax")
xmax.text = str(int(shape["points"][1][0]))
ymax = ET.SubElement(bndbox, "ymax")
ymax.text = str(int(shape["points"][1][1]))
return ET.tostring(root, encoding="unicode")
current_dir = os.getcwd()
parent_dir = os.path.abspath(os.path.join(current_dir, os.pardir))
output_dir = os.path.join(parent_dir, "xml_output")
if not os.path.exists(output_dir):
os.makedirs(output_dir)
for file in os.listdir(current_dir):
if file.endswith(".json"):
with open(os.path.join(current_dir, file), "r") as json_file:
json_data = json.load(json_file)
xml_data = convert_json_to_xml(json_data)
xml_filename = file.replace(".json", ".xml")
output_path = os.path.join(output_dir, xml_filename)
with open(output_path, "w") as xml_file:
xml_file.write(xml_data)
print("所有JSON文件已经转换为XML并保存在文件夹 xml_output 中")
|
