yolov5+shufflenet轻量化目标检测

一、backbone与head

yolov5中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, [1024, 3, 2]],  # 7-P5/32
   [-1, 3, C3, [1024]],
   [-1, 1, SPPF, [1024, 5]],  # 9
  ]

from：表示与上层连接

number：模块重复次数

modul：模块类型

args：卷积层参数信息

其中，存在“#0-P1/2，# 1-P2/4..”标记，

"0","1"..表示当成层的序号，在head模块中用到并且+1

“P1/2,P2/4”表示经过该层，feature map的尺寸缩减为原图2倍、4倍。。

其中C3模块结构形式：