简单介绍:
yolov5s.yaml文件解读
一、前导知识-yaml文件介绍
二、查看配置文件
yolov5的所有网络yaml文件如下:
我们需要修改的网络参数配置文件为yolov5s.yaml
,其文件的内容如下:
1. 参数配置
- depth_multiple:channel的缩放系数,就是将配置里面的backbone和head部分有关通道的设置,全部乘以该系数即可
- width_multiple:BottleneckCSP模块的层缩放系数,将所有的BottleneckCSP模块的number系数乘上该参数就可以最终的层个数
2.anchors配置
3、backbone
- from: 表示当前模块的输入来自哪一层的输出,-1表示来自上一层的输出,层编号由0开始计数。
- number: 表示当前模块的理论重复次数,实际的重复次数还要由上面的参数depth_multiple共同决定,该参数影响整体网络模型的深度。
- module: 模块类名(也可以理解为模块的功能),通过这个类名在common.py中寻找相应的类,进行模块化的网络搭建。
- args: 是一个list,对应到模块类搭建时需要的参数,主要是channel、kernel_size、stride、padding、bias等
4、head
三、修改配置文件
主要修改backbone
python yolo.py
,you will see: