YoloV5网络架构
总体架构
- Backbone(主体) : Focus, BottleneckCSP, CSP
- Head : PANet + Dectect (Yolov3/Yolov4 Head)
代码部分
# parameters
nc: 80 # number of classes 类别数
depth_multiple: 0.33 # model depth multiple 控制模型的深度 (BottleneckCSP数)
width_multiple: 0.50 # layer channel multiple 控制Conv通道个数 (卷积核数量)
# depth_multiple表示BottleneckCSP模块的层缩放因子,将所有的BottleneckCSP模块的Bottleneck乘上该参数得到最终个数。
# width_multiple表示卷积通道的缩放因子,就是将配置里面的backbone和head部分有关Conv通道的设置,全部乘以该系数。
# 通过这两个参数就可以实现不同复杂度的模型设计。
# anchors
anchors: # 先验框 或者 锚框
- [10,13, 16,30, 33,23] # P3/8 代表stride=8 也就是8倍下采样尺度后先验框的大小 宽度是10 长度是 13
- [30,61, 62,45, 59,119] # P4/16
- [116,90, 156,198, 373,326] # P5/32
# YOLOv5 backbone
backbone:
# [from, number, module, args]
# from列参数:当前模块输入来自哪一层输出;-1 代表是从上一层获得的输入
# number列参数:本模块重复次数;1表示只有一个,3表示有三个相同的模块
[[-1, 1, Focus, [64, 3]], # 0-P1/2
[-1, 1, Conv,

本文详细介绍了YoloV5的网络架构,包括Backbone的Focus、BottleneckCSP和CSP模块,以及Head部分的PANet和Detect。通过调整depth_multiple和width_multiple参数,可以定制模型的复杂度。网络结构中,FOCUS用于下采样,池化增大感受野,而PANet则通过上采样和特征融合增强信息。此外,还探讨了网络深度、卷积核数量对模型性能的影响。
最低0.47元/天 解锁文章
1027

被折叠的 条评论
为什么被折叠?



