YOLOv7训练自己的数据集(超详细）

置顶 Mr Dinosaur

已于 2023-01-13 11:13:14 修改

阅读量2w

点赞数 63

分类专栏： YOLOv7 文章标签：深度学习人工智能计算机视觉目标检测 python

于 2022-07-13 14:52:01 首次发布

本文链接：https://blog.csdn.net/qq_58355216/article/details/125677147

版权

YOLOv7 专栏收录该内容

2 篇文章

订阅专栏

介绍

2022年7月，YOLOv7来临，论文链接：https://arxiv.org/abs/2207.02696

代码链接：

GitHub - WongKinYiu/yolov7: Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors

废话不多说，赶紧上车！

文件配置

1、数据集

自己创建一个myself.yaml文件用来配置路径，路径格式与之前的V5、V6不同，只需要配置txt路径就可以

train-list.txt和val-list.txt文件里存放的都是图片的绝对路径（也可以放入相对路径）

如何获取图像的绝对路径，脚本写在下面了（也可以获取相对路径）

# From Mr. Dinosaur

import os


def listdir(path, list_name):  # 传入存储的list
    for file in os.listdir(path):
        file_path = os.path.join(path, file)
        if os.path.isdir(file_path):
            listdir(file_path, list_name)
        else:
            list_name.append(file_path)


list_name = []
path = 'D:/PythonProject/data/'  # 文件夹路径
listdir(path, list_name)
print(list_name)

with open('./list.txt', 'w') as f:  # 要存入的txt
    write = ''
    for i in list_name:
        write = write + str(i) + '\n'
    f.write(write)