UAVDT数据集标签转为yolo格式

UAVDT数据集是以XML格式存储的，而Yolo格式是以txt文件存储的。因此，需要将UAVDT数据集中的XML标签转换为Yolo格式的txt文件。以下是将UAVDT数据集标签转为Yolo格式的步骤：

1. 首先需要下载UAVDT数据集，并解压缩。

2. 创建一个Python脚本，用于将XML标签转换为Yolo格式的txt文件。

3. 在脚本中导入必要的库，如xml.etree.ElementTree和os。

4. 使用os.listdir()函数来获取UAVDT数据集中所有的图像文件名，并遍历每个图像文件。

5. 对于每个图像文件，使用xml.etree.ElementTree库中的parse()函数来解析XML标签文件，并获取每个目标的类别、边界框坐标等信息。

6. 将类别转换为数字格式，并计算出中心点坐标和边界框的宽度和高度。

7. 根据Yolo格式的要求，将目标信息存储在一个txt文件中，并将该文件保存在与图像文件同名的文件夹中。

下面是一个简单的Python脚本，用于将UAVDT数据集标签转换为Yolo格式的txt文件：

import xml.etree.ElementTree as ET
import os

def convert_labels(label_dir, image_dir, output_dir):
    # 获取所有图像文件名
    image_files = os.listdir(image_dir)
    for image_file in image_files:
        # 获取标签文件名
        label_file = image_file.replace('.jpg', '.xml')
        label_path = os.path.join(label_dir, label_file)
        # 解析标签文件
        tree = ET.parse(label_path)
        root = tree.getroot()
        # 遍历每个目标
        for obj in root.findall('object'):
            # 获取类别
            class_name = obj.find('name').text
            if class_name == 'car':
                class_id = 0
            elif class_name == 'bus':
                class_id = 1
            elif class_name == 'truck':
                class_id = 2
            # 获取边界框坐标
            bbox = obj.find('bndbox')
            x_min = float(bbox.find('xmin').text)
            y_min = float(bbox.find('ymin').text)
            x_max = float(bbox.find('xmax').text)
            y_max = float(bbox.find('ymax').text)
            # 计算中心点坐标和宽度、高度
            x = (x_min + x_max) / 2.0
            y = (y_min + y_max) / 2.0
            w = x_max - x_min
            h = y_max - y_min
            # 将目标信息写入txt文件
            txt_file = image_file.replace('.jpg', '.txt')
            txt_path = os.path.join(output_dir, txt_file)
            with open(txt_path, 'a') as f:
                f.write(f"{class_id} {x} {y} {w} {h}\n")

在上述脚本中，convert_labels()函数接受三个参数：标签文件夹路径、图像文件夹路径和输出文件夹路径。该函数会将每个图像文件的标签转换为一个txt文件，并将该文件保存在输出文件夹中。其中，类别名称car、bus和truck分别转换为数字0、1和2。如果UAVDT数据集中有其他类别，需要根据实际情况进行修改。