hutao1030813002的博客

用Python对我们自己标注的数据集转化为YOLO训练需要的txt文件 一. 数据分类 在项目的根目录下新建一个maketxt.py文件。 该脚本会在straw/ImageSets文件夹下生成：trainval.txt, test.txt, train.txt, val.txt，内容如下图所示，将图片分成了训练集，验证集和测试集，并将文件名（不带扩展名）汇总在txt文件中。 import os import random trainval_percent = 0.1 train_percent = 0.9

用Python对数据集进行重命名 新建一个rename.py 文件 import os pic_path = "F:\\straw\\" def rename(): piclist = os.listdir(pic_path) total_num = len(piclist) i = 1 for pic in piclist: if pic.endswith(".jpg"): old_path = os.path.join(os