使用yolov3训练的数据集

1， 建立VOC2007文件夹，按照原来的文件夹名称新建

2， 将图片按照00000x统一命名，保存在JPEGImages文件夹中

3，工具给图片打标签
https://pan.baidu.com/s/1eoUOFVyz7AuB79h_sRgo5Q labelImg，密码：gyf3。

下载文件后，打开…/data/predefined_classes.txt文件，可以发现这里都是图片标签——把你将要用到的标签都事先存入在这里，注意标签不能有中文。每次使用都把.exe、data这两个文件拖到桌面上（如果直接在文件夹内运行.exe会报错不能运行），打开labelImg.exe文件，运行界面如下：就可以开始给图片打标签了

Ctrl+s 保存

d 下一张图片

w 建立选框

4，在VOC2007下新建test.py文件用来生成ImageSets/Main/4个文件

import os

import
random

 

trainval_percent
= 0.1

train_percent
= 0.9

xmlfilepath
= 'Annotations'

txtsavepath
= '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, tr)

 

ftrainval
= open('ImageSets/Main/trainval.txt', 'w')

ftest =
open('ImageSets/Main/test.txt', 'w')

ftrain =
open('ImageSets/Main/train.txt', 'w')

fval =
open('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.