将YOLO格式的txt标注改成voc的xml标注格式，txt转xml

 txt格式的标注文件转voc的xml文件：

   txt格式(123.jpg pig x_min y_min x_max y_max)

import copy
from lxml.etree import Element, SubElement, tostring, ElementTree
import cv2
 
# 修改为你自己的路径
template_file = 'C:\\Users\\zzw\\Desktop\\jongjv\\txt-xml\\000.xml'
target_dir = 'C:\\Users\\zzw\\Desktop\\jongjv\\txt-xml\\Annotations\\'
image_dir = 'C:\\Users\\zzw\\Desktop\\jongjv\\txt-xml\\images\\'  # 图片文件夹
train_file = 'C:\\Users\\zzw\\Desktop\\jongjv\\txt-xml\\123.txt'  # 存储了图片信息的txt文件
 
with open(train_file) as f:
    trainfiles = f.readlines()  # 标注数据 格式(123.jpg pig x_min y_min x_max y_max)
 
file_names = []
tree = ElementTree()
 
for line in trainfiles:
    trainFile = line.split()
    file_name = trainFile[0]
    print(file_name)
 
    # 如果没有重复，则顺利进行。这给的数据集一张图片的多个框没有写在一起。
    if file_name not in file_names:
        file_names.append(file_name)
        lable = trainFile[1]
        
        #因为要使用labelimg来编辑xml，所以将trainFile[2] float类型转成整型。再将整型转成str类型存到xml文件里面。
        xmin = str(int(float(trainFile[2])))
        ymin = str(int(float(trainFile[3])))
        xmax = str(int(float(trainFile[4])))
        ymax = str(int(float(trainFile[5])))
 
        tree.parse(template_file)
        root = tree.getroot()
        root.find('filename').text = file_name
 
        # size
        sz = root.find('size')
        im = cv2.imread(image_dir + file_name)#读取图片信息
 
        sz.find('height').text = str(1080)
        sz.find('width').text = str(1920)
        sz.find('depth').text = str(3)
 
        # object 因为我的数据集都只有一个框
        obj = root.find('object')
 
        obj.find('name').text = lable
        bb = obj.find('bndbox')
        bb.find('xmin').text = xmin
        bb.find('ymin').text = ymin
        bb.find('xmax').text = xmax
        bb.find('ymax').text = ymax
        # 如果重复，则需要添加object框
    else:
        lable = trainFile[1]
        
        xmin = str(int(float(trainFile[2])))
        ymin = str(int(float(trainFile[3])))
        xmax = str(int(float(trainFile[4])))
        ymax = str(int(float(trainFile[5])))
 
        xml_file = file_name.replace('jpg', 'xml')
        tree.parse(target_dir + xml_file)#如果已经重复
        root = tree.getroot()
 
        obj_ori = root.find('object')
 
        obj = copy.deepcopy(obj_ori)  # 注意这里深拷贝
 
        obj.find('name').text = lable
        bb = obj.find('bndbox')
        bb.find('xmin').text = xmin
        bb.find('ymin').text = ymin
        bb.find('xmax').text = xmax
        bb.find('ymax').text = ymax
        root.append(obj)
 
    xml_file = file_name.replace('jpg', 'xml')
    tree.write(target_dir + xml_file, encoding='utf-8')