YOLOv5网络结构+代码+应用详解|CSDN创作打卡

深度学习入门小菜鸟,希望像做笔记记录自己学的东西,也希望能帮助到同样入门的人,更希望大佬们帮忙纠错啦~侵权立删。

这一篇文章属于是对YOLOv5系列的一个整理汇总。

目录

一、输入端分析

二、整体网络结构图以及总括介绍

三、Backbone

四、Neck

 五、Head

六、代码实现网络具体构建

七、利用YOLOv5训练自己的数据集

八、YOLOv5训练结果分析


一、输入端分析

YOLOv5输入端(一)—— Mosaic数据增强|CSDN创作打卡_tt丫的博客-CSDN博客

YOLOv5输入端(二)|CSDN创作打卡_tt丫的博客-CSDN博客(自适应锚框计算+自适应图片缩放)


二、整体网络结构图以及总括介绍

网络结构图如下图所示:

(其中CBL的标注都是有关输出的)

在这里感谢博友352(兄弟你前面那个符号是啥我这边显示不出来,不好意思)的提醒

v5.0版本(下图中的CBL改为CBS)

目前更新到v6.0版本

(这里比较大的改动是:用CBS代替 Focus 层,主要是为了方便模型导出 )

YOLOv5网络结构分为3个部分,Backbone(主干部分),Neck和Head。

Backbone作用:特征提取

Neck作用:对特征进行一波混合与组合,并且把这些特征传递给预测层

Head作用:进行最终的预测输出


三、Backbone

Backbone主要由Focus,CBS,BottleneckCSP(/C3)以及SPP等组成。

🌳Focus作用:进行一个下采样,并且减少计算量加快网络速度。(v5.0)

🌳CBS作用:(就是我们常说的卷积啦,虽然这里的卷积不是直接意义上的nn.Conv2d。)获取特征。

有关Focus和CBS的分析可以康康博主之前写的博客

YOLOv5中的Focus层详解_tt丫的博客-CSDN博客

🌳BottleneckCSP(/C3)作用:残差结构,学习更多的特征。

有关BottleneckCSP(/C3)的分析可以康康博主之前写的博客

YOLOv5中的CSP结构_tt丫的博客-CSDN博客

🌳SPP/SPPF作用:空间金字塔池化,将前面得到的任意大小的特征图转换成固定大小的特征向量。实现局部特征和全局特征的featherMap级别的融合。(感谢@qq_44622851博友的提醒)

(v5.0是SPP,v6.0是SPPF)

有关SPP/SPPF的分析可以康康博主之前写的博客

YOLOv5中的SPP/SPPF结构详解_tt丫的博客-CSDN博客_spp yolov5


四、Neck

Neck说白了就是FPN+PAN结构的设计。

关于FPN和PAN可以看博主之前写的文章

深度学习中的FPN详解_tt丫的博客-CSDN博客

深度学习之PAN详解_tt丫的博客-CSDN博客

(这里v5.0和v6.0的区别请对照上图,懒得改了嘻嘻)


 五、Head

我理解的Head就是输出端。如果单从网络结构上来说的话,他就是对Neck出来的3个输出分别进行卷积操作。然而Head的主要部分在于它的输出推理及处理。具体的可以看博主之前的文章,里面有关于损失函数,NMS以及源码的解析。

YOLOv5的输出端详解_tt丫的博客-CSDN博客


六、代码实现网络具体构建

详见YOLOv5的模型构建源码详解|CSDN创作打卡_tt丫的博客-CSDN博客


七、利用YOLOv5训练自己的数据集

YOLOv5训练自己的数据集详解_tt丫的博客-CSDN博客


八、YOLOv5训练结果分析

YOLOv5训练结果性能分析_tt丫的博客-CSDN博客_yolov5训练结果分析


欢迎大家在评论区批评指正,谢谢大家~

评论 20
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

tt丫

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值