yolov5制作自己的数据集时的格式转换和数据划分

最新推荐文章于 2024-08-20 11:00:37 发布

Lininggggggg

最新推荐文章于 2024-08-20 11:00:37 发布

阅读量553

点赞数

分类专栏：黑板项目文章标签： python

本文链接：https://blog.csdn.net/Lininggggggg/article/details/124040608

版权

本文介绍了使用YoloV5框架创建自定义数据集时，如何进行数据格式转换和数据划分的过程，详细阐述了具体步骤，并提供了一个参考链接。

摘要由CSDN通过智能技术生成

yolov5制作自己的数据集时的格式转换和数据划分

参考链接：https://blog.csdn.net/llh_1178/article/details/114528795

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=["blackboard"]


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('VOCdevkit/VOC2007/Annotations/%s.xml' %image_id)
    out_file = open('VOCdevkit/VOC2007/YOLOLabels/%s.txt' %image_id, 'w')
    # in_file = open('./Annotations/%s.xml' %image_id)
    # out_file = open('./labels/%s.txt' %image_id, 'w')

    tree=ET.parse(in_file)
    root = tree.getroot()
    size = root.find('size'