模型整体框架
对应代码文件 yolov5 -> models -> .yaml文件
不同yaml文件的depth_multiple、width_multiple不同,通过这两个参数控制模型大小
模型从小到大依次为 n -> s -> m -> l -> x
示例:yolov5s.yaml
module
-
Conv——ConvBNSiLU(Convolution, Batch normalization,SiLU)
- Activation——SiLU(Sigmoid Linear Unit)
- x ∗ s i g m o i d ( x ) = x / ( e − x + 1 ) x * sigmoid(x) = x / (e ^ {-x} + 1) x
- Activation——SiLU(Sigmoid Linear Unit)