yolov7训练自己的数据集(超详细)(代码片段)

MrDinosaur MrDinosaur     2023-04-12     490

关键词:

介绍

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)

2、train.py

官网下载模型文件,train.py文件只支持YOLOv7和YOLOv7-X模型

train文件还是和V5一样,为了方便,我将需要用到的文件放在了根目录下

路径修改完之后右击运行即可

3、train_aux.py

如果你想使用较大的预训练模型,请使用train_aux.py进行训练,否则效果会很差(本人亲测)

下面放上对比图:(上面V7,下面V5)

--weights

下载位置在官网的GitHub上(我是用的是yolov7-d6-training.pt)

--cfg

请使用cfg-training-中的模型文件

 --hyp

文件夹data-hyp.scratch.p6.yaml

 运行train_aux.py

效果对比

在此放上YOLOv7和YOLOv5的对比图:(左V7,右V5)

报错解决

YOLOv7 训练报错 IndexError: list index out of range_Mr Dinosaur的博客-CSDN博客https://blog.csdn.net/qq_58355216/article/details/125842647?spm=1001.2014.3001.5501

 评价

无论是训练的速度、还是精度、召回和map的提升,V7的表现都是十分显著的,称得上是YOLO界的扛把子,期待作者之后的优化和更新。

YOLOv8发布

官网的YOLOv8最近开始发布,效果要比V7更快更准,训练和测试方法已写好,感兴趣的小伙伴快动手操作一下 

YOLOv8训练自己的数据集https://blog.csdn.net/qq_58355216/article/details/128671030?spm=1001.2014.3001.5501

yolov7训练自己的数据集(代码片段)

yolov7训练自己的数据集一.安装yolov7并配置1.安装Anaconda2.创建虚拟环境3.下载yolov7源码4.下载权重5.安装依赖二.运行目标检测三.训练自的数据集1.创建所需文件2.下载labelImg标注工具3.配置训练的相关文件四.正式训练五.使用自己的... 查看详情

yolov7训练自己的数据集(口罩检测)(代码片段)

YOLOv7训练自己的数据集(口罩检测)前言前提条件实验环境项目结构制作自己的数据集数据集目录结构训练自己的数据集VOC格式数据集转换成YOLO格式数据集修改cfg配置新建一个myyolov7.yaml配置文件myyolov7.yaml内容创建自己... 查看详情

yolo|用yolov7训练自己的数据集(超详细版)

...程)二.克隆项目配置库gitclonehttps://github.com/WongKinYiu/yolov7打开项目下载要求的库:cdyolov7 查看详情

最新版yolov6训练自己的数据集(超详细完整版!)(代码片段)

接着上篇文章继续写,本篇文章讲如何训练自己的数据集。从官网下载YOLOv6源码:meituan/YOLOv6:YOLOv6:asingle-stageobjectdetectionframeworkdedicatedtoindustrialapplications.(github.com)https://github.com/meituan/YOLOv6一、创建文件将t 查看详情

win10安装yolov7训练自己的数据集(代码片段)

1.安装https://www.pudn.com/news/62ddf431864d5c73acfb1a09.html下载yolov7源码yolov7修改requirements.txt修改torch以及torchvision版本,这里直接指定好版本让其版本对应安装pipinstall-rrequirements.txt-fhttps://download.pytorch.org/w 查看详情

win10安装yolov7训练自己的数据集(代码片段)

1.安装https://www.pudn.com/news/62ddf431864d5c73acfb1a09.html下载yolov7源码yolov7修改requirements.txt修改torch以及torchvision版本,这里直接指定好版本让其版本对应安装pipinstall-rrequirements.txt-fhttps://download.pytorch.org/w 查看详情

[软件工具]yolov3yolov4yolov7训练自己的数据集windows上超简单全程不写代码

darknet训练自己的数据集一般要写代码自己转化数据集,还要修改配置文件,数据集也不保证都是正确标注的,这个流程一般对初学者或者不熟悉的人经常弄错,只要一个配置参数改错就会导致训练出现问题,... 查看详情

yolox训练自己的数据集(超详细)

...些库文件        (3)、安装apex文件二、YOLOX训练自己的数据集1、数据集预处理2、修改代码(1ÿ 查看详情

yolov7pytorch(代码片段)

yolov7主干部分结构图:yolov7主干yolov7数据集处理代码:yolov7数据集处理代码yolov7训练参数解释:yolov7训练参数【与本文代码有区别】yolov7训练代码详解:yolov7训练代码详解目录训练自己的训练集训练自己的训练集... 查看详情

yolov7训练自己的数据集及报错处理(代码片段)

...#################D:\\Anaconda3\\envs\\py38torch_gpu\\python.exeD:\\needed\\yolov7-main\\train.py--weightsweights/yolov7.pt--cfgcfg/training/yolov7.yaml--datadata/datasets.yaml--device0--batch-size8--epoch5 YOLOR 2022-9-16torch1.9.0+cu111CUDA:0(NVIDIAGeForceRTX3060Ti,8191.5MB)Namespace(adam=... 查看详情

yolov7:从搭配环境到训练自己的数据集(代码片段)

一.下载源码在github中下载Yolov7源码并解压缩YOLOv7源码数据标注可以参考这篇文章https://blog.csdn.net/qq_39779449/article/details/110172948?spm=1001.2014.3001.5502标签格式二.配置环境整体配置树结构安装Yolov7gpu环境在Anconda中创造Yolov7环境安装... 查看详情

[课程][原创]yolov7训练自己数据集实例分割模型

课程地址:搞定系列:yolov7-segmention训练实例分割模型win版--深度学习视频教程-人工智能-CSDN程序员研修院课程介绍课程目录讨论留言你将收获学会如何搭建自己cuda环境学会如何搭建自己yolov7环境学会如何使用roboflow或者l... 查看详情

yolov5训练自己的数据集(超详细,小白必看)

...f0c;接下来看文件内需要放哪些内容。说一下,既然是训练自己的数据集就需要我们对自己寻找的数据集进行标注,这边我使用的是labelimg它是用Python编写的,并将Qt用于其图形界面具体怎么使用大家可以参考(21条消息)... 查看详情

darknet-yolov3训练自己的数据集windows上超简单全程不写代码(代码片段)

darknet-yolov3训练自己的数据集一般要写代码自己转化数据集,还要修改配置文件,数据集也不保证都是正确标注的,这个流程一般对初学者或者不熟悉的人经常弄错,只要一个配置参数改错就会导致训练出现问题&#... 查看详情

深度学习100例|第53天:用yolov5训练自己的数据集(超级详细完整版)(代码片段)

...#xff09;配置完YOLOv5需要的环境后,今天我们试着用YOLOv5训练自己的数据。࿰ 查看详情

visdrone数据集visdrone+yolov7结果记录(代码片段)

VisDrone+YOLOv7结果YOLOv7训练验证集测试集YOLOv7训练命令python-mtorch.distributed.launch--nproc_per_node4--master_port9527train.py--workers4--device0,1,2,3--sync-bn--batch-size32--datadata/VisDrone.yaml--img640640--cfgcfg/training/yolov7.yaml--nameyolov7_--hypdata/hyp.scratch.p5.yam... 查看详情

yolov5-pytorch版-windows下训练自己的数据集,内含voc批量转yolo方法。(自称宇宙超级巨详细步骤)(代码片段)

Yolov5-Pytorch版-Windows下训练自己的数据集(自称宇宙超级巨详细步骤)一.准备数据集1.在labelimg下直接生成yolo的.txt格式2.Pytorch版批量.xml格式转yolo的.txt格式3.数据集制作二.Yolov5实现训练1.修改两个.yaml文件2.修改train.py文件3... 查看详情

yolov7(目标检测)入门教程详解---检测,推理,训练(代码片段)

目录一.前言二.yolov7源码下载三.detect(检测)四.Train(训练)数据准备:labellmg:配置训练的相关文件 配置数据集文件正式训练:推理:推理效果:五.总结一.前言  上篇文章:YOLOv7(目标... 查看详情