yolov5训练自己的VOC数据集

一.VOC数据准备

 

voc数据集目录结构 ：

               ----voc

                           ----Annotations

                           ----ImageSets

                                        ----Main

                           ----JPEGImages

在根目录下新建makeTXT.py，将数据集划分，并且在Main文件夹下构建4个TXT：train.txt，test.txt，trainval.txt，val.txt。代码如下：

import os
import random
 
trainval_percent = 0.1
train_percent = 0.9
xmlfilepath = 'voc/Annotations'
txtsavepath = 'voc/ImageSets'
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, tr)
 
ftrainval = open('voc/ImageSets/Main/trainval.txt', 'w')
ftest = open('voc/ImageSets/Main/test.txt', 'w')
ftrain = open('voc/ImageSets/Main/train.txt', 'w')
fval = open('data/ImageSets/Main/val.txt', 'w')
 
for i in list:
    name = total_xml[i][:-4] + '\n'
    if i in trainval:
        ftrainval.write(name)
        if i in train:
            ftest.write(name)
        else:
            fval.write(name)
    else:
        ftrain.write(name)
 
ftrainval.close()
ftrain.close()
fval.close()
ftest.close()

 在根目