python制作训练集_Yolov5训练自建数据集

环境安装暂不缀述

1.数据准备

本人将数据放在了yolo-master项目下自建了coco目录，目录结构如下：

val2017.txt:保存验证集的图片(绝对路径，相对路径可能会报错)

train2017.txt:保存训练集的图片(绝对路径)

labels:保存val和train下图片的ground truth

images:保存val和train的图片

其中，labels下目录结构如下，与images目录相同，只不过labels/val2017和labels/train2017分别对应images/val2017和images/train2017下每张图片的标签(这个标签是经过归一化的)：

1.1 生成val2017.txt和train2017.txt的代码如下：

即生成记录图片路径的txt文件。

import os

image_path = './coco/images/val2017/'#修改为自己的路径

file = open('./coco/val2017.txt', 'w')#修改