Computer and Modernization ›› 2023, Vol. 0 ›› Issue (01): 37-42.

Previous Articles     Next Articles

SAR Ship Classification Based on Multi-convolutional Neural Network Fusion

  

  1. (1. Aerospace Information Research Institute, Chinese Academy of Sciences, Beijing 100190, China; 2. School of Electronic, Electrical and Communication Engineering, University of Chinese Academy of Sciences, Beijing 100049, China)
  • Online:2023-03-02 Published:2023-03-02

Abstract: The accuracy of small ship classification in Syntactic Aperture Radar (SAR) images is low. To solve the problem, a classification approach based on the weighted fusion of different convolutional neural network results is proposed. Firstly, a high-resolution convolutional neural network is constructed to conduct multi-scale feature fusion, fine-tuning model and label smoothing are introduced to reduce the problem of training over-fitting. Then three single classification models are trained using the high-resolution network, MobileNetv2 network and SqueezeNet network. Finally, the results of three classification models are fused by weighted voting. The fusion method is used to carry out classification experiment on GF-3 ship dataset, the results obtained are: precision 94.83%, recall rate 95.43%, F1 score 0.9513. Experimental results show that the algorithm model proposed in this paper has better classification ability, which verifies its effectiveness in high-resolution SAR image ship classification.

Key words: SAR images; high-resolution convolutional neural networks; fine tuning model, label smooth; weighted voting; ship classification