Pytorch实战 |Y3 yolov5s.yaml文件解读

yolo5s.yaml文件解读
摘要由CSDN通过智能技术生成
  • Yolov5配置了4种不同大小的网络模型,分别是YOLOv5s、YOLOv5m、YOLOv5l、YOLOv5x,其中YOLOv5s是网络深度和宽度最小但检测速度最快的模型,其他3种模型都是在YOLOv5的基础上不断加深、加宽网络是的网络规模扩大,在增强模型检测性能的同时,增加了计算资源和速度。处于对检测精度、模型大小以及检测速度的综合考量,本文对YOLOv5s进行研究。
  • ./models/yolov5s.yaml文件是yolov5s的网络结构,如果要改算法网络结构可以再此文件中进行修改,然后再修改./models/common.py和./models/yolo.py中的代码。

一、YOLOv5s网络结构

如下图,根据yolov5s.yaml文件中结构,参考网上资料使用PPT画的。
在这里插入图片描述

二、代码解读

1、参数配置

nc: 80  # number of classes
depth_multiple: 0.33  # model depth multiple  控制模型的深度
width_multiple: 0.50  # layer channel multiple 控制Conv通道channel个数
  • depth_multiple 表示 BottleneckCSP 模块层的缩放因子,将所有的 BottleneckCSP 模块的 Bottleneck 乘上该参数得到最终个数。控制子模块数量=int(number*depth)
  • width_multiple 表示卷积通道的缩放因子,就是将配置里面的 backbone 和 head 部分有关 Conv 通道的设置,全部乘以该系数。控制卷积核的数量=int(number*width)

YOLOv5s、YOLOv5m、YOLOv5l、YOLOv5x四种模型的区别仅在于depth_multiple 和width_multiple两个参数的不同。
以上三个参数,会用于模型搭建yolo.py文件中

读取 yaml 中的 anchors 和 parameters
anchors, nc, gd, gw = d['anchors'], d['nc'], d['depth_multiple'], d['width_multiple'] 

通过深度参数 depth gain, 在搭建每层时, 实际深度 = 理论深度(每一

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值