import xml.etree.ElementTree as ET
import pickle
import os
from os import listdir, getcwd
from os.path import joinclasses = ["dog", "cat"]def convert(size, box):dw = 1./(size[0])dh = 1./(size[1])x = (box[0] + box[1])/2.0 - 1y = (box[2] + box[3])/2.0 - 1w = box[1] - box[0]h = box[3] - box[2]x = x*dww = w*dwy = y*dhh = h*dhreturn (x,y,w,h)def convert_annotation(rootpath,xmlname):xmlpath = rootpath + '/labels'xmlfile = os.path.join(xmlpath,xmlname)with open(xmlfile, "r") as in_file:txtname = xmlname[:-4]+'.txt'txtpath = rootpath + '/labelYOLOs'if not os.path.exists(txtpath):os.makedirs(txtpath)txtfile = os.path.join(txtpath,txtname)with open(txtfile, "w+") as out_file:tree=ET.parse(in_file)root = tree.getroot()size = root.find('size')w = int(size.find('width').text)h = int(size.find('height').text)out_file.truncate()for obj in root.iter('object'):difficult = obj.find('difficult').textcls = obj.find('name').textif cls not in classes or int(difficult)==1:continuecls_id = classes.index(cls)xmlbox = obj.find('bndbox')b = (float(xmlbox.find('xmin').text), float(xmlbox.find('xmax').text), float(xmlbox.find('ymin').text), float(xmlbox.find('ymax').text))bb = convert((w,h), b)out_file.write(str(cls_id) + " " + " ".join([str(a) for a in bb]) + '\n')if __name__ == "__main__":rootpath='/home/aimhabo/YOLOv3SMP/data'xmlpath=rootpath+'/labels'list=os.listdir(xmlpath)for i in range(0,len(list)) :path = os.path.join(xmlpath,list[i])if ('.xml' in path)or('.XML' in path):convert_annotation(rootpath,list[i])print('done', i)else:print('not xml file',i)