首页 > 其他分享 >手把手教你调参 YOLO v6(train.py)

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

时间:2023-01-29 12:12:18浏览次数:37  
标签:YOLOv6 py 训练 -- YOLO 默认 train v6

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

官网网址:

YOLO v6
https://github.com/meituan/YOLOv6

train.py参数分析
打开文件tools->train.py

 

1、介绍

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

2、--data-path

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

 

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

正确的YOLO格式yaml文件:

将数据集A放在根目录:

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

3、--conf-file

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

 

 

 

4.--img-size

 

 

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

5.--batch-size

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

6、--epochs

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

7、--workers

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

8、--device

 

 

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

9、--eval-interval

 

 

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

 

10、--eval-final-only

 

 

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

11、--heavy-eval-range

 

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

12、--check-images

 

 

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

13、--check-labels

 

 

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

14、--output-dir

 

 

训练时输出的路径

 

15、--name

 

 

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

 

16、--dist_url

 

 

默认的网址;不需要修改

17、--gpu_count

 

 

GPU数量类型;默认整型

18、--local_rank

 

 

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

19、--resume

 

 

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

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

最新版YOLOv6训练自己的数据集(超详细完整版!)
https://blog.csdn.net/qq_58355216/article/details/125525243?spm=1001.2014.3001.5502

YOLOv6训练时报错解决方法
https://blog.csdn.net/qq_58355216/article/details/125552713?spm=1001.2014.3001.5502
下面是测试的教程:

手把手教你运行YOLOv6(超详细)
https://blog.csdn.net/qq_58355216/article/details/125497521?spm=1001.2014.3001.5502

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

标签:YOLOv6,py,训练,--,YOLO,默认,train,v6
From: https://www.cnblogs.com/kn-zheng/p/17072328.html

相关文章

  • Python 的垃圾回收机制【译】
    几乎所有的高级编程语言都有自己的垃圾回收机制,开发者不需要关注内存的申请与释放,Python也不例外。Python官方团队的文章https://devguide.python.org/internals/garba......
  • yolov5检测框重合重复,手动调参方法(调整detect,val的conf,iou)
    一、问题描述:检测框重复出现上述问题一般是整体检测方向没错,但conf-thres和iou-thres的参数需要调整。(在默认值0.25和0.45的基础上,提高置信区间,降低iou)conf-thres:置信......
  • yolov5 优化——mosaic相关
    概述Mosaic利用了四张图片重新排布成一张图片,根据论文所说其拥有一个巨大的优点是丰富检测物体的背景:随机缩放增加了很多小目标,让网络的鲁棒性更好;且在BN计算的时候一下子......
  • Python工具箱系列(二十三)
    基于游标得操作游标是数据库操作的相对底层的能力。简单的操作如下:importmysql.connectorimportrandomhost='localhost'user='root'password='8848is8848'......
  • yolov5 提速多GPU训练显存低的问题
    修改前:按照配置,在train.py配置如下:运行pythontrain.py后nvidia-smi显示显存占用如下:修改后参考yolov5官方中的issue中,有人提到的分布式多进程的方法:在yolov5运行......
  • Python字符串中用于转义的字符很多
    Python字符串中用于转义的字符很多,这里列举了几个比较常用的几个,更多的转义应用会放在合集的文章里。\n换行符:可以在一行内创建多行输出的字符串;\t制表符:相当于四个空格......
  • 【Python视频下载】Python字符串连接的5种方法
    导读在使用Python的时候,经常会进行字符串操作,本文总结了一下Python字符串连接的5种方法,希望对大家有所帮助。1.加号第一种,有编程经验的人,估计都知道很多语言里面是用加号连......
  • 关于yolov5训练后权重较大的原因及其解决方案
    yolov5在官方预训练权重的基础之上再训练后权重较预训练权重大很多,为了大家更加直观的了解为什么及其解决方案,此篇简单介绍一下。精度变化官方给的预训练权重是FP16,而我们......
  • python functools.singledispatch的使用,针对函数不同的传参类型,做出不同的操作。
    背景希望函数根据传参类型不同,做出不同的操作.根据传入变量的类型来判断需要输出的内容,常见的做法是把这个函数做成一个分派函数,在这个函数中通过大量的if/elif/else来判......
  • Python工具箱系列(二十四)
    不管多少人黑微软,微软出品的大多数产品都能够深入人心,成为精品。在数据库领域,微软为专业人士提供SQLServer(简称mssql)。为日常办公人士提供Access与Excel这两款数据存储与......