Yolov5-images_tag.py-voc_to_yolo.py-data.yaml

images_tag.py
# 该脚本文件需要修改第11-12行，设置train、val、test的切分的比率
import os
import random
import argparse

parser = argparse.ArgumentParser()
parser.add_argument('--xml_path', default='E:\A\Yolov5\data\dataset\VOCdevkit\Annotations', type=str, help='input xml label path')
parser.add_argument('--txt_path', default='E:\A\Yolov5\data\dataset\VOCdevkit\Imagesets', type=str, help='output txt label path')
opt = parser.parse_args()

trainval_percent = 0.9
train_percent = 0.7 #这里的train_percent 是指占trainval_percent中的
xmlfilepath = opt.xml_path
txtsavepath = opt.txt_path
total_xml = os.listdir(xmlfilepath)
if not os.path.exists(txtsavepath):
os.makedirs(txtsavepath)

num = len(total_xml)
list_index = range(num)
tv = int(num * trainval_percent)