《计算机应用与软件杂志》发表论文赏析
作者:李卓阳, 郭荣佐, 车思超, 杨军
摘要:边缘设备的资源有限,而深度神经网络模型规模庞大且参数量众多。针对基于深度神经网络的合成孔径雷达(Synthetic Aperture Radar, SAR)图像舰船目标检测模型在边缘设备上部署难度大的问题,提出一种基于剪枝的压缩方案,对 YOLOv3-SPP 模型进行轻量化。使用该方法在 AIR-SARShip-1.0 数据集上进行实验。实验结果表明,采用的压缩方案能将原始模型压缩到 10% 以下,而速度提升了 1.5 倍,精度值只是略微降低了 0.02。这大大降低了模型在边缘端的部署难度,突破了硬件资源的限制。
关键词:合成孔径雷达, 舰船目标检测, 压缩, 过滤器剪枝, 层剪枝, 蒸馏