yolov5 xml转txt

要将XML格式转换为YOLOv5所需的TXT格式，可以使用以下代码进行处理:

import os
import xml.etree.ElementTree as ET

def convert_xml_to_txt(xml_path, txt_path):
    # 打开XML文件
    tree = ET.parse(xml_path)
    root = tree.getroot()

    with open(txt_path, 'w') as txt_file:
        for obj in root.findall('object'):
            # 获取对象的类别和边界框坐标
            name = obj.find('name').text
            xmin = int(obj.find('bndbox/xmin').text)
            ymin = int(obj.find('bndbox/ymin').text)
            xmax = int(obj.find('bndbox/xmax').text)
            ymax = int(obj.find('bndbox/ymax').text)

            # 计算中心点坐标和边界框宽高
            x_center = (xmin + xmax) / 2
            y_center = (ymin + ymax) / 2
            width = xmax - xmin
            height = ymax - ymin

            # 将数据写入TXT文件
            line = f"{name} {x_center} {y_center} {width} {height}"
            txt_file.write(line + '\n')

    print(f"Successfully converted {xml_path} to {txt_path}")

# 指定XML文件和输出TXT文件的路径
xml_file = 'path/to/input.xml'
txt_file = 'path/to/output.txt'

# 转换XML到TXT
convert_xml_to_txt(xml_file, txt_file)

请将`path/to/input.xml`替换为输入XML文件的路径，将`path/to/output.txt`替换为输出TXT文件的路径。运行该代码后，将会生成对应的TXT文件，其中包含了每个对象的类别和边界框信息，符合YOLOv5的要求。

请注意，此代码仅适用于单个XML文件转换。如果需要批量转换，请使用适当的循环来处理多个文件。

中提供的XML示例中包含了一个对象的类别和边界框信息。你可以根据自己的XML文件结构进行相应的调整以匹配代码中的解析逻辑。

希望以上信息对你有所帮助！如果有任何进一步的问题，请随时提问。