YOLOv5网络结构完全解读【源码+手绘网络结构+模块结构】

YOLOv5网络结构详解

前言

虽然寒假前就用YOLOv5跑完好几个数据集了,但是一直没有深究其网络结构及特点。开学后的一个多星期里也是饱食终日,无所用心。所以三天前给自己立了一个flag:读懂YOLOv5的网络结构并记录自己的学习过程。于是才有了这几天的浅显的、一知半解的YOLO详解。列位看官阅读过程中有问题欢迎评论区交流。

总体网络结构

YOLOv5网络结构完全解读【源码+手绘网络结构+模块结构】_第1张图片

1 YOLO的anchor设定

首先,对anchor的产生及检测过程进行了分析,给出了可以带入到自己数据集的anchor源码。
YOLO的anchor设定

2 YOLO的Backbone

从yolo的网络配置文件入手,细致分析了backbone的构成及特征图的传播过程,并分析了各个部件的组成,如CSP、CBS、SSPF、Bottleneck,对于瓶颈模块Bottleneck,以Resnet引入,给出了resnet和CPS的区别。
YOLOv5的Backbone详解

3 YOLO的Neck

经典的FPN+PAN设计,即自顶向下+自下而上的特征提取方式。
YOLOv5的Neck端设计

4 YOLO的head

主要是Detect模块的理解
YOLOv5的head详解

你可能感兴趣的:(YOLOv5,计算机视觉,目标检测,深度学习,pytorch)