首页|SAR图像舰船检测的神经网络关联剪枝方法

SAR图像舰船检测的神经网络关联剪枝方法

扫码查看
合成孔径雷达(Synthetic Aperture Radar,SAR)由于其具备全天时、全天候的工作特点,使其在海洋环境监测、海洋资源调查和海洋防灾减灾等领域得到了广泛的应用.其中,基于SAR图像的舰船目标检测是SAR图像处理中的重要部分,其在军用和民用领域均具有重要的意义.本文针对基于深度学习实现的SAR图像目标检测算法参数计算量大、内存占用率高的问题提出了关联剪枝方法.该方法通过对网络进行改进,将相关联的卷积同时进行剪枝,并在训练结束后统一映射到低维度上以实现剪枝操作.通过在SSDD、SAR-Ship-Data-set和HRSID上进行实验,可以在保证平均精度(AP50)下降小于2%的前提下,针对FCOS网络实现70%以上的剪枝率,验证了所提方法的有效性.
Neural Network Associative Pruning for Ship Detection in SAR Images
The all-day and all-weather working characteristics of synthetic aperture radar(SAR)have made it widely used in the fields of marine environment monitoring,marine resources survey and marine disaster prevention and mitigation.Ship target detection based on SAR images is a critical element of SAR image processing,which is of great significance in both military and civilian fields.In this paper,a deep learning based associative pruning method is pro-posed to solve the problem of large parameter computation and high memory consumption of SAR image target detection algorithm.By improving the network,the method prunes the associated convolutions simultaneously,and maps them to the lower dimension after the training to realize the pruning operation.Through experiments on SSDD,SAR-Ship-Data-set and HRSID,it is possible to achieve a pruning rate to more than 70%for FCOS networks under the premise of ensur-ing that the average precision(AP50)decreases by less than 2%,which verifies the effectiveness of the proposed method.

synthetic aperture radar(SAR)model pruningobject detectiondeep learning

张丽丽、王贤俊、屈乐乐、刘雨轩

展开 >

沈阳航空航天大学电子信息工程学院,辽宁沈阳 110136

合成孔径雷达 模型剪枝 目标检测 深度学习

辽宁省教育厅项目辽宁省教育厅项目

LJKZ0174LJKMZ20220533

2024

雷达科学与技术
中国电子科技集团公司第38研究所 中国电子学会无线电定位技术分会

雷达科学与技术

CSTPCD北大核心
影响因子:0.665
ISSN:1672-2337
年,卷(期):2024.22(3)
  • 22