VOC标注格式转换成YOLO标注格式

最新推荐文章于 2025-02-23 12:58:52 发布

bigcindy

最新推荐文章于 2025-02-23 12:58:52 发布

阅读量6.8k

点赞数 9

分类专栏： YOLOv5 文章标签： VOC YOLO 标注格式目标检测标签文件

本文链接：https://blog.youkuaiyun.com/Jwenxue/article/details/107913655

版权

import xml.etree.ElementTree as ET
import pickle
import os
from os import listdir, getcwd
from os.path import join

# 这里就体现出来了咱们在1.2步骤的时候我说的尽量按照那个目录名进行操作的优势，
# 在这可以剩下很多去修改名称的精力
# sets=[('2012', 'train'), ('2012', 'val'), ('2007', 'train'), ('2007', 'val'), ('2007', 'test')]
sets=[ ('2007', 'train'), ('2007', 'val'), ('2007', 'test')]  # 我只用了VOC2007

classes = ["aeroplane", "bicycle", "bird", "boat", "bottle",
           "bus", "car", "cat", "chair", "cow",
           "diningtable", "dog", "horse", "motorbike", "person",
           "pottedplant", "sheep", "sofa", "train", "tvmonitor"]
#classes = ["face"]  # 修改为自己的label

def convert(size, b