An Efficient and Lightweight CNN Model With Soft Quantification for Ship Detection in SAR Images
Authors:
- 24 June 2022
- journal article
- research article
- Published by Institute of Electrical and Electronics Engineers (IEEE) in IEEE Transactions on Geoscience and Remote Sensing
- Vol. 60 (01962892) , 1-13
Abstract
Convolutional neural networks (CNNs) have been widely used for synthetic aperture radar (SAR) target detection. Typical methods based on CNN have obtained favorable detection accuracy at the cost of high model complexity, and thus are difficult to be directly applied to real-time satellites on board as well as maritime rescue. To deal with this problem, this article proposes an efficient and lightweight target detection network incorporating soft quantization. First, to compensate for the lack of accuracy caused by lightweight networks, a feature fusion module called split bidirectional feature pyramid network is proposed to alleviate the interference of complex background on SAR images. Meanwhile, to adapt the lightweight network and the feature fusion module, a linear transformation module is presented to enhance the linear representation of the model via learnable parameters. Eventually, to make the model size smaller, a soft quantization algorithm is proposed to reduce the accuracy degradation caused by quantization errors. We validate the robustness of the model in several publicly available datasets. Experimental results show that our model achieves 97.0% detection accuracy on SAR ship detection dataset, with a 0.9% accuracy improvement compared to mainstream methods using less than $15\times $ the number of parameters and less than $6\times $ the number of flops.
Keywords
Funding Information
- National Natural Science Foundation of China (61976166)
- Key Research and Development Program of Shaanxi (2021GY-030)
- Fundamental Research Funds for the Central Universities (JB210115)
- Open Research Projects of Laboratory of Pinghu
This publication has 48 references indexed in Scilit:
- Target Detection via Bayesian-Morphological Saliency in High-Resolution SAR ImagesIEEE Transactions on Geoscience and Remote Sensing, 2017
- An Intensity-Space Domain CFAR Method for Ship Detection in HR SAR ImagesIEEE Geoscience and Remote Sensing Letters, 2017
- Synthetic Aperture Radar Ship Detection Using Haar-Like FeaturesIEEE Geoscience and Remote Sensing Letters, 2016
- XNOR-Net: ImageNet Classification Using Binary Convolutional Neural Networks2016
- You Only Look Once: Unified, Real-Time Object Detection2016
- Fast R-CNN2015
- Detection of Moving Ships Based on a Combination of Magnitude and Phase in Along-Track Interferometric SAR—Part II: Statistical Modeling and CFAR DetectionIEEE Transactions on Geoscience and Remote Sensing, 2015
- Detection of Moving Ships Based on a Combination of Magnitude and Phase in Along-Track Interferometric SAR—Part I: SIMP Metric and Its PerformanceIEEE Transactions on Geoscience and Remote Sensing, 2015
- Compressed-Domain Ship Detection on Spaceborne Optical Image Using Deep Neural Network and Extreme Learning MachineIEEE Transactions on Geoscience and Remote Sensing, 2014
- Rich Feature Hierarchies for Accurate Object Detection and Semantic Segmentation2014