面向小目标检测的轻量化YOLOv3算法

作者：张官荣; 陈相; 赵玉; 王建军; 易国彪

来源：激光与光电子学进展, 2021, 1-13.

• YOLOv3
• 稀疏训练
• 模型剪枝

为了提高面向遥感图像目标检测的YOLOv3-CS算法的检测速度，本文提出了基于BN层γ参数的自适应稀疏因子调整算法，以γ作为通道的重要性判断依据对YOLOv3-CS进行剪枝，得到YOLOv3-CSP目标检测模型。实验结果表明，本文剪枝方法在mAP损失0.22%的情况下，对YOLOv3-CS的模型大小压缩95.92%，检测速度提高173%；本文提出的YOLOv3-CSP可以应用于检测精度和实时性要求较高的场合。

• 单位
  西北工业大学; 空军工程大学