清华Tinghua100K交通标志数据集.json标签转.xml（python代码）

清华Tinghua100K交通标志数据集.json标签转.xml

数据集连接：Tinghua100K
转成与labelImg标注生成文件相同的XML格式（位置仅保留了bbox）

import json
import xml.dom.minidom
import cv2

img_path='F:/tt100k_dataset/data/train/' #train图片路径
annos = json.loads(open("F:/tt100k_dataset/data/annotations.json").read())
xml_path = 'F:/tt100k_dataset/data/xml_train/'#xml保存路径

for line in open(img_path+"ids.txt"):
    img_name=line.replace('\n','')
    img_file=img_name + '.jpg'
    img = cv2.imread(img_path + img_file)
    sp = img.shape
    img_height = str(sp[0])  # height(rows) of image
    img_width = str(sp[1])

    doc = xml.dom.minidom.Document()
    # creat a root node which name is annotation
    annotation = doc.createElement('annotation')
    # add the root node to the dom document object
    doc.appendChild(annotation)

    # add the folder subnode
    folder = doc.createElement('folder')
    folder_text = doc.createTextNode('JPEGImages')
    folder.appendChild(folder_text)
    annotation.appendChild(folder)

    # add the filename subnode
    filename = doc.createElement('filename')
    filename_text = doc