yoloV3实验文档.tar.gz

import os
import random

trainval_percent = 0.1
train_percent = 0.9
xmlfilepath = ‘data/Annotations‘
txtsavepath = ‘data/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（‘data/ImageSets/trainval.txt‘ ‘w‘）
ftest = open（‘data/ImageSets/test.txt‘ ‘w‘）
ftrain = open（‘data/ImageSets/train.txt‘ ‘w‘）
fval = open（‘data/ImageSets/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（）