【Week-Y3】修改yolov5s的backbone的第4层和第6层中C3的理论循环次数

2024-03-29 09:28

本文主要是介绍【Week-Y3】修改yolov5s的backbone的第4层和第6层中C3的理论循环次数,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

修改backbone中C3的循环次数

  • 一、yolov5s.yaml文件解析
  • 二、训练
  • 三、总结

  • 🍨 本文为🔗365天深度学习训练营 中的学习记录博客
  • 🍖 原作者:K同学啊 | 接辅导、项目定制

📕本次任务:将yolov5s网络模型中第4层的C32改为C31,第6层的C33修改为C32
📕提示:仅需修改./models/yolov5s.yaml文件

与v5m、v5l、v5x相比,yolov5s是网络深度和宽度最小但检测速度最快的模型。

yolov5s.yaml中,depth_multiple表示模型子模块的数量,代表了模型的深度,width_multiple表示卷积层的通道个数,代表了卷积核个数,通过修改这两个参数的值就能设计不同复杂度的yolo结构。

yolov5syolov5myolov5lyolov5x四种结构的差异在于这两个参数的不同。

yolov5s.yaml
|-depth_multiple: 0.33
|-width_multiple: 0.5yolov5m.yaml
|-depth_multiple: 0.67
|-width_multiple: 0.75yolov5l.yaml
|-depth_multiple: 1.0
|-width_multiple: 1.0yolov5x.yaml
|-depth_multiple: 1.33 
|-width_multiple: 1.25 

一、yolov5s.yaml文件解析

yolov5-master的文件结构如下:

yolov5-master
|-classify
|-data
|-models (本次学习需要的文件在这里面)|-hub|-segment|-common.py|-experi,emtal.py|-tf.py...|-yolov5s.yaml (这是本次学习需要修改的文件,文件内容是yolov5s的网络结构)|-yolov5m.yaml|-yolov5l.yaml|-yolov5x.yaml
|-runs
|-segment
|-utils
...

打开yolov5s.yaml文件,内容包括参数、backbone结构和head结构,其中,在backbone部分第4层和第6层(从0开始)中C3的数量已按本次学习任务进行了修改:

# YOLOv5 🚀 by Ultralytics, AGPL-3.0 license
# https://github.com/ultralytics/yolov5# Parameters
nc: 80 # number of classes
depth_multiple: 0.33 # model depth multiple 控制模型深度,也就是控制子模块数量(与本次学习有关)
width_multiple: 0.50 # layer channel multiple 控制Conv通道channel个数(卷积核个数)
anchors:- [10, 13, 16, 30, 33, 23] # P3/8  检测小目标,10,13是一组,一共检测三组小目标- [30, 61, 62, 45, 59, 119] # P4/16  检测中目标- [116, 90, 156, 198, 373, 326] # P5/32  检测大目标# YOLOv5 v6.0 backbone
backbone:# [from, number, module, args][[-1, 1, Conv, [64, 6, 2, 2]], # 0-P1/2[-1, 1, Conv, [128, 3, 2]], # 1-P2/4[-1, 3, C3, [128]],  # 2[-1, 1, Conv, [256, 3, 2]], # 3-P3/8[-1, 3, C3, [256]],   # 4 这里改为C3*1,原来是C3*2[-1, 1, Conv, [512, 3, 2]], # 5-P4/16[-1, 6, C3, [512]],  # 6 这里改为C3*2,原来是C3*3[-1, 1, Conv, [1024, 3, 2]], # 7-P5/32[-1, 3, C3, [1024]],  # 8[-1, 1, SPPF, [1024, 5]], # 9]# YOLOv5 v6.0 head
head: [[-1, 1, Conv, [512, 1, 1]],  # 10[-1, 1, nn.Upsample, [None, 2, "nearest"]],[[-1, 6], 1, Concat, [1]], # cat backbone P4[-1, 3, C3, [512, False]], # 13[-1, 1, Conv, [256, 1, 1]],[-1, 1, nn.Upsample, [None, 2, "nearest"]],[[-1, 4], 1, Concat, [1]], # cat backbone P3[-1, 3, C3, [256, False]], # 17 (P3/8-small)[-1, 1, Conv, [256, 3, 2]],[[-1, 14], 1, Concat, [1]], # cat head P4[-1, 3, C3, [512, False]], # 20 (P4/16-medium)[-1, 1, Conv, [512, 3, 2]],[[-1, 10], 1, Concat, [1]], # cat head P5[-1, 3, C3, [1024, False]], # 23 (P5/32-large)[[17, 20, 23], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)]

yolov5s的网络结构图:
在这里插入图片描述
控制端输出的网络结构:
在这里插入图片描述

二、训练

数据集:水果数据集 【使用week Y2的数据集】
命令行:python .\train.py --img 900 --batch 2 --epoch 100 --data .\fruit.yaml --cfg .\models\yolov5s.yaml --weights .\yolov5s.pt --device cpu【与week Y2的训练命令是一样的】

训练结果如下,总共花费训练时间3.066hr:
在这里插入图片描述
训练的类别结果如下:
在这里插入图片描述
所有类别的F1曲线图如下:
在这里插入图片描述

三、总结

(1)比较了本次学习和【week Y2】学习所打印的训练网络结构,本次学习打印出的网络layers为200,而【week Y2】学习所打印的层数为214,说明修改有效,但是为什么显示的网络结构是一样的?

这篇关于【Week-Y3】修改yolov5s的backbone的第4层和第6层中C3的理论循环次数的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/858224

相关文章

Python实现无痛修改第三方库源码的方法详解

《Python实现无痛修改第三方库源码的方法详解》很多时候,我们下载的第三方库是不会有需求不满足的情况,但也有极少的情况,第三方库没有兼顾到需求,本文将介绍几个修改源码的操作,大家可以根据需求进行选择... 目录需求不符合模拟示例 1. 修改源文件2. 继承修改3. 猴子补丁4. 追踪局部变量需求不符合很

Linux修改pip和conda缓存路径的几种方法

《Linux修改pip和conda缓存路径的几种方法》在Python生态中,pip和conda是两种常见的软件包管理工具,它们在安装、更新和卸载软件包时都会使用缓存来提高效率,适当地修改它们的缓存路径... 目录一、pip 和 conda 的缓存机制1. pip 的缓存机制默认缓存路径2. conda 的缓

Linux修改pip临时目录方法的详解

《Linux修改pip临时目录方法的详解》在Linux系统中,pip在安装Python包时会使用临时目录(TMPDIR),但默认的临时目录可能会受到存储空间不足或权限问题的影响,所以本文将详细介绍如何... 目录引言一、为什么要修改 pip 的临时目录?1. 解决存储空间不足的问题2. 解决权限问题3. 提

Python循环缓冲区的应用详解

《Python循环缓冲区的应用详解》循环缓冲区是一个线性缓冲区,逻辑上被视为一个循环的结构,本文主要为大家介绍了Python中循环缓冲区的相关应用,有兴趣的小伙伴可以了解一下... 目录什么是循环缓冲区循环缓冲区的结构python中的循环缓冲区实现运行循环缓冲区循环缓冲区的优势应用案例Python中的实现库

Linux文件名修改方法大全

《Linux文件名修改方法大全》在Linux系统中,文件名修改是一个常见且重要的操作,文件名修改可以更好地管理文件和文件夹,使其更具可读性和有序性,本文将介绍三种在Linux系统下常用的文件名修改方法... 目录一、引言二、使用mv命令修改文件名三、使用rename命令修改文件名四、mv命令和rename命

mybatis-plus 实现查询表名动态修改的示例代码

《mybatis-plus实现查询表名动态修改的示例代码》通过MyBatis-Plus实现表名的动态替换,根据配置或入参选择不同的表,本文主要介绍了mybatis-plus实现查询表名动态修改的示... 目录实现数据库初始化依赖包配置读取类设置 myBATis-plus 插件测试通过 mybatis-plu

Linux下修改hostname的三种实现方式

《Linux下修改hostname的三种实现方式》:本文主要介绍Linux下修改hostname的三种实现方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录linux下修改ho编程stname三种方式方法1:修改配置文件方法2:hFvEWEostnamectl命

Git如何修改已提交人的用户名和邮箱

《Git如何修改已提交人的用户名和邮箱》文章介绍了如何修改Git已提交人的用户名和邮箱,包括注意事项和具体步骤,确保操作正确无误... 目录git修改已提交人的用户名和邮箱前言第一步第二步总结git修改已提交人的用户名和邮箱前言需注意以下两点内容:需要在顶层目录下(php就是 .git 文件夹所在的目

Java嵌套for循环优化方案分享

《Java嵌套for循环优化方案分享》介绍了Java中嵌套for循环的优化方法,包括减少循环次数、合并循环、使用更高效的数据结构、并行处理、预处理和缓存、算法优化、尽量减少对象创建以及本地变量优化,通... 目录Java 嵌套 for 循环优化方案1. 减少循环次数2. 合并循环3. 使用更高效的数据结构4

Python中顺序结构和循环结构示例代码

《Python中顺序结构和循环结构示例代码》:本文主要介绍Python中的条件语句和循环语句,条件语句用于根据条件执行不同的代码块,循环语句用于重复执行一段代码,文章还详细说明了range函数的使... 目录一、条件语句(1)条件语句的定义(2)条件语句的语法(a)单分支 if(b)双分支 if-else(