Yolov5的yaml文件解读

        

本文为[365天深度学习训练营](https://mp.weixin.qq.com/s/0dvHCaOoFnW8SCp3JpzKxg)中的学习记录客
原作者:[K同学啊]https://mtyjkh.blog.youkuaiyun.com/

        最新Ultralytics给了Yolov5的两个网络模型,分别是yolov5和yolov5-p6两个部分,本文选择yolov5作为研究对象进行介绍。

        yolov5.yaml文件是yolov5网络结构的定义文件,像改进算法的需要在该文件修该相关参数,让后再修改common.py和yolo.py的相关文件。

1.参数配置

        这里面nc是分类的数目,scales是不同模型的信息,包括深度,宽度和最大通道数目

2.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]]
  - [-1, 1, Conv, [256, 3, 2]] # 3-P3/8
  - [-1, 6, C3, [256]]
  - [-1, 1, Conv, [512, 3, 2]] # 5-P4/16
  - [-1, 9, C3, [512]]
  - [-1, 1, Conv, [768, 3, 2]] # 7-P5/3
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值