Small-Sample Sonar Image Classification Based on Deep Learning
Zezhou Dai, Liang Hong, Tong Duan
Abstract
Deep learning is a core technology for sonar image classification. However, owing to the cost of sampling, a lack of data for sonar image classification impedes the training and deployment of classifiers. Classic deep learning models such as AlexNet, VGG, GoogleNet, and ResNet suffer from low recognition rates and overfitting. This paper proposes a novel network (ResNet-ACW) based on a residual network and a combined few-shot strategy, which is derived from generative adversarial networks (GAN) and transfer learning (TL). We establish a sonar image dataset of six-category targets, which are formed by sidescan sonar, forward-looking sonar, and three-dimensional imaging sonar. The training process of ResNet-ACW on the sonar image dataset is more stable and the classification accuracy is also improved through an asymmetric convolution and a designed network structure. We design a novel GAN (LN-PGAN) that can generate images more efficiently to enhance our dataset and fine-tune ResNet-ACW pretrained on mini-ImageNet. Our method achieves 95.93% accuracy and a 14.19% increase in the six-category target sonar image classification tasks.