手把手教你调参 YOLO v6(train.py)

距离YOLOv6发布已经有一周的时间了,中途修复了好多bug,一起来看看吧

官网网址:

YOLO v6 手把手教你调参 YOLO v6(train.py)https://github.com/meituan/YOLOv6 ;

train.py参数分析

打开文件tools->train.py

手把手教你调参 YOLO v6(train.py)

1、介绍

手把手教你调参 YOLO v6(train.py)

没啥用,担心我们不知道这是YOLOv6的train文件….

2、–data-path

手把手教你调参 YOLO v6(train.py)

存放数据集路径的yaml,作者提供了一个coco和一个yolo格式训练的yaml

手把手教你调参 YOLO v6(train.py)

注意:v6不需要使用images文件夹,将以前images中的train和val文件夹图片与labels文件放在同一个目录就可以了,不然容易找不到数据集,这个bug官方还未更新,更新了请提醒我(7月1日)

正确的YOLO格式yaml文件:

将数据集A放在根目录:

手把手教你调参 YOLO v6(train.py)

还有一点,yolo v6目前使用单类别数据集训练时,会出现一系列的报错,尽量使用多类别,这个bug官方目前还没修复,虽然他们说支持单类别…

3、–conf-file

手把手教你调参 YOLO v6(train.py)

用于指定模型配置文件,根据自己的需求进行配置;v6的配置文件中使调参又方便了一些

手把手教你调参 YOLO v6(train.py)

手把手教你调参 YOLO v6(train.py)

4.–img-size

手把手教你调参 YOLO v6(train.py)

训练时图片的大小,默认是640*640,根据自己需求进行调整

5.–batch-size

手把手教你调参 YOLO v6(train.py)

这个根据自己电脑显存进行调节,尽量越大越好

6、–epochs

手把手教你调参 YOLO v6(train.py)

训练的次数,并不是越大越好,300左右基本就可以了

7、–workers

手把手教你调参 YOLO v6(train.py)

默认8就好,如果自己电脑硬件不支持就填0,越接近0,训练的时候就会越占用CPU

8、–device

手把手教你调参 YOLO v6(train.py)

显卡的使用,默认就是使用第一张卡

9、–eval-interval

手把手教你调参 YOLO v6(train.py)

更新间隔评估;这个参数是真的坑,默认每20次更新一次,导致很多人的map都是0,还以为是程序的问题,坑了一大片人,官网论坛都在问,官方也没说是这里的情况

手把手教你调参 YOLO v6(train.py)

10、–eval-final-only

手把手教你调参 YOLO v6(train.py)

在最后一个批次进行评估;默认是true,不需要进行改动

11、–heavy-eval-range

手把手教你调参 YOLO v6(train.py)

在最后50个epochs时进行持续更新,可以和–eval-interval 一起使用

12、–check-images

手把手教你调参 YOLO v6(train.py)

训练前检查图像数据集,默认true

13、–check-labels

手把手教你调参 YOLO v6(train.py)

训练前检查标签数据集,默认true

14、–output-dir

手把手教你调参 YOLO v6(train.py)

训练时输出的路径

手把手教你调参 YOLO v6(train.py)

15、–name

手把手教你调参 YOLO v6(train.py)

训练数据保存到的文件名称

手把手教你调参 YOLO v6(train.py)

16、–dist_url

手把手教你调参 YOLO v6(train.py)

默认的网址;不需要修改

17、–gpu_count

手把手教你调参 YOLO v6(train.py)

GPU数量类型;默认整型

18、–local_rank

手把手教你调参 YOLO v6(train.py)

单机多卡训练;基本用不到

19、–resume

手把手教你调参 YOLO v6(train.py)

断点训练,是否在之前训练的一个模型基础上继续训练;需要接着之前训练就改为true

总结:

目前YOLOv6的程序还不够完善,训练时会遇到各种问题,我做了个总结,希望能帮助到你们

最新版YOLOv6训练自己的数据集(超详细完整版!)手把手教你调参 YOLO v6(train.py)https://blog.csdn.net/qq_58355216/article/details/125525243?spm=1001.2014.3001.5502 ;

YOLOv6训练时报错解决方法手把手教你调参 YOLO v6(train.py)https://blog.csdn.net/qq_58355216/article/details/125552713?spm=1001.2014.3001.5502 ;下面是测试的教程:

手把手教你运行YOLOv6(超详细)手把手教你调参 YOLO v6(train.py)https://blog.csdn.net/qq_58355216/article/details/125497521?spm=1001.2014.3001.5502 ;

如果对你有帮助,请给个点赞关注支持吧!

手把手教你调参 YOLO v6(train.py)

持续更新中 …

Original: https://blog.csdn.net/qq_58355216/article/details/125558843
Author: Mr Dinosaur
Title: 手把手教你调参 YOLO v6(train.py)

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/686745/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球