请给我写一个将voc格式的数据集转换成yolo格式的数据集的代码

今天药忘吃喽~ 2024-03-27 12:31 88阅读 0赞

在这里给你提供一个简单的 Python 代码，用于将 VOC 格式的数据集转换为 YOLO 格式的数据集。

首先，我们需要导入一些必要的库：

import os
    import xml.etree.ElementTree as ET

然后，我们可以读取 VOC 数据集的标注信息，并将其转换为 YOLO 格式的标注信息。

# 读取 VOC 格式的标注信息
    tree = ET.parse(voc_annotation_file)
    root = tree.getroot()
    
    # 遍历标注信息
    for obj in root.iter('object'):
        difficult = obj.find('difficult').text
        cls = obj.find('name').text
        if cls not in classes or int(difficult) == 1:
            continue
        cls_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))
        # 转换为 YOLO 格式的标注信息
        bb = convert_to_yolo_format(b, width, height)
        out_file.write(str(cls_id) + " " + " ".join([str(a) for a in bb]) + '\n')

最后，我们还需要一个用于将 VOC 格式的边界框转换为 YOLO 格式的边界框的函数：

def convert_to_yolo_format(b, width, height):
        xmin, xmax, ymin, ymax = b
        x = (xmin + xmax) / 2 / width
        y = (ymin + ymax) / 2 / height
        w = (xmax - xmin) / width
        h = (ymax - ymin) / height
        return (x, y, w, h)

这样就可以将 VOC 格式的数据集转换为 YOLO 格式的数据集了。希望这个代码能帮助你！