voc To yolov5-6.1数据集格式转换

已有的数据集

操作

第一步：划分训练集、验证集、测试集

本章节讲解的只有划分：训练集、验证集。

通过脚本文件（createImageSet.py）生成训练集和验证集

createImageSet.py代码

import os
import random

trainval_percent = 0.9
train_percent = 0.8
xmlfilepath = './safe-helmet/Safety Helmet Wearing Dataset/Annotations'
txtsavepath = './safe-helmet/Safety Helmet Wearing Dataset/ImageSets/Main'
total_xml = os.listdir(xmlfilepath)

num = len(total_xml)
list = range(num)
tv = int(num * trainval_percent)
tr = int(tv * train_percent)
trainval = random.sample(list, tv)
train = random.sample(trainval