yolov5训练时如何将xml文件转为.txt文件

在使用 YOLOv5 训练时，我们需要将标注信息从 XML 文件中提取出来并转换成 TXT 文件，以便训练时可以读取到标注信息。

以下是将 XML 文件转换为 YOLOv5 可以读取的 TXT 文件的步骤：

1. 确认 XML 文件的格式是否符合 YOLOv5 的要求。YOLOv5 要求每个 XML 文件中只包含一个 object，并且需要包含以下信息：object 类别、bbox 的左上角坐标(xmin, ymin)、bbox 的右下角坐标(xmax, ymax)。

2. 编写 Python 脚本，使用 ElementTree 库来解析 XML 文件，并将标注信息写入 TXT 文件。具体实现可以参考以下代码：

import xml.etree.ElementTree as ET

def convert_annotation(xml_file, txt_file):
    tree = ET.parse(xml_file)
    root = tree.getroot()
    with open(txt_file, 'w') as f:
        for obj in root.iter('object'):
            cls = obj.find('name').text
            bbox = obj.find('bndbox')
            xmin = bbox.find('xmin').text
            ymin = bbox.find('ymin').text
            xmax = bbox.find('xmax').text
            ymax = bbox.find('ymax').text
            f.write(f'{cls} {xmin},{ymin},{xmax},{ymax}\n')

其中，xml_file 是输入的 XML 文件名，txt_file 是输出的 TXT 文件名。

3. 调用 convert_annotation 函数，将 XML 文件转换为 TXT 文件。可以使用以下代码进行测试：

xml_file = 'path/to/xml/file.xml'
txt_file = 'path/to/txt/file.txt'
convert_annotation(xml_file, txt_file)

这样就可以将 XML 文件转换为 YOLOv5 可以读取的 TXT 文件了。