yolov5训练数据集划分
按照默认8:1:1划分训练集,测试集,验证集。
txt文件出现在imageset文件夹。
import os
import random
trainval_percent = 0.9
train_percent = 0.9
xmlfilepath = 'data/Annotations'
txtsavepath = 'data/ImageSets'
total_xml = os.listdir(xmlfilepath)
num = len(total_xml