YOLOV5-3-数据集格式转化及训练集和验证集划分

最新推荐文章于 2025-12-04 10:55:27 发布

原创

最新推荐文章于 2025-12-04 10:55:27 发布 · 279 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#YOLO #人工智能 #深度学习

VOC标签格式转yolo格式并划分训练集和测试集

用labelIMG打标的数据是不满足yolov5直接训练的，需要分为两个文件夹------训练集和数据集

同时xml格式的标签也需要更改txt格式。先上代码。

import xml.etree.ElementTree as ET
import pickle
import os
from os import listdir, getcwd
from os.path import join
import random
from shutil import copyfile
 
classes = ["hat", "person"]
#classes=["ball"]
 
TRAIN_RATIO = 80
 
def clear_hidden_files(path):
    dir_list = os.listdir(path)
    for i in dir_list:
        abspath = os.path.join(os.path.abspath(path), i)
        if os.path.isfile(abspath):
            if i.startswith("._"):
                os.remove(abspath)
        else:
            clear_hidden_files(abspath)
 
def convert(size, box):
    dw = 1./size[0]
    dh = 1./size[1]
    x = (box[0] + box[1])/2.0
    y = (box[2] + box[3])/2.0
    w = box[1] - box[0]
    h = box[3] - box[2]
    x = x*dw
    w = w*dw
    y = y*dh
    h = h*dh
    return (x,y,w,h)
 
def convert_annotation(image_id):
    in_file = open('VOC