本文为[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