基于剪枝的 SAR 图像舰船目标检测模型压缩方案

COMPRESSION SCHEME OF SAR IMAGES SHIP TARGET DETECTION MODEL BASED ON PRUNING

  • 摘要: 边缘设备的资源有限,而深度神经网络模型规模庞大且参数量众多。针对基于深度神经网络的合成孔径雷达(Synthetic Aperture Radar, SAR)图像舰船目标检测模型在边缘设备上部署难度大的问题,提出一种基于剪枝的压缩方案,对 YOLOv3-SPP 模型进行轻量化。使用该方法在 AIR-SARShip-1.0 数据集上进行实验。实验结果表明,采用的压缩方案能将原始模型压缩到 10% 以下,而速度提升了 1.5 倍,精度值只是略微降低了 0.02。这大大降低了模型在边缘端的部署难度,突破了硬件资源的限制。

     

    Abstract: The resources of edge devices are limited, but the deep neural network model is large and has many parameters. Aimed at the difficulty of deploying the SAR image ship target detection model based on deep neural network on edge devices, a compression scheme based on pruning is proposed to lightweight the YOLOv3-SPP model. Experiments were performed on AIR-SARSHIP-1.0 data set using this method. Experimental results show that the original model can be compressed to less than 10% by the proposed compression scheme, while the speed is increased by 1.5 times and the accuracy is only slightly decreased by 0.02, which greatly reduces the difficulty of deploying the model at the edge and breaks through the limitations of hardware resources.

     

/

返回文章
返回