Veri-776数据集处理 xml文件提取转换成txt文件

最近在做重识别工作，数据处理时需要进行文件格式转换，这里将代码记录下来

from os import getcwd
import os
from xml.dom.minidom import parse
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

vehicleids = []

def convert_annotation():
    in_file = open('dataset/VeRi/train_label.xml')
    list_file = open('train_list.txt','w')
    tree = parse(in_file)

    root = tree.documentElement

    for item in root.getElementsByTagName("Item"):  
        if item.hasAttribute("imageName"):
            name = item.getAttribute("imageName")
        if item.hasAttribute("vehicleID"):
            vehicleid = item.getAttribute("vehicleID")
            if vehicleid not in vehicleids :
                vehicleids.append(vehicleid)
            vid = vehicleids.index(vehicleid)
        if item.hasAttribute("colorID"):
            colorid = int (item.getAttribute("colorID"))
        if item.hasAttribute("typeID"):
            typeid = int (item.getAttribute("typeID"))
        list_file.write(name + " " + str(vid)  + " " + str(colorid) + " " +str(typeid) + "\n")

    list_file.close()



if __name__ == '__main__':
	convert_annotation()  #imagename vehiclenum colorid typeid