Acta Geodaetica et Cartographica Sinica ›› 2021, Vol. 50 ›› Issue (2): 260-269.doi: 10.11947/j.AGCS.2021.20200187

• Marine Survey • Previous Articles     Next Articles

The transfer learning with convolutional neural network method of side-scan sonar to identify wreck images

TANG Yulin, JIN Shaohua, BIAN Gang, ZHANG Yonghou, LI Fan   

  1. Department of Hydrography and Cartography, Dalian Naval Academy, Dalian 116018, China
  • Received:2020-05-12 Revised:2020-10-14 Published:2021-03-03
  • Supported by:
    The National Natural Science Foundation of China (Nos. 41876103;41576105)

Abstract: The Side-scan sonar image automatic recognition is an important part of verification for underwater obstacle and wreck search and rescue, in view of the traditional artificial interpretation of side-scan sonar image is inefficient, time consuming and resource consumption and strong subjective uncertainty and excessive reliance on experience. This paper attempts to introduce the method of convolutional neural network, considering that the side-scan sonar shipwreck image belongs to a small sample data set, and an automatic recognition method of side-scan sonar shipwreck image based on transfer learning is proposed.The sample data were expanded by means of normalization and image enhancement, the training set and testing set were divided into 4∶1 proportions, and an improved model was designed according to the characteristics of the side-scan sonar wreck data set by referring to the classical VGG-16 model, then, the improved model trained on the ImageNet image data set is used to learn and experiment on the small sample side-scan sonar shipwreck data set using two transfer learning methods: freeze and train and fine-tuning, and compared with new learning. The results show that the accuracy of the three methods for the recognition of side-scan sonar shipwreck images is 93.71%, 84.49% and 90.58%, respectively. The first transfer learning method has the highest accuracy rate, the fastest model convergence speed, and the highest AP value 92.45%, which is 8.06% and 3.06% higher than the second transfer learning and the new learning method, respectively,and has a better effect in improving the model’s recognition ability and training efficiency. which verifies the effectiveness and feasibility of this method and has certain practical guiding significance.

Key words: side-scan sonar wreck image, image recognition, transfer learning, convolutional neural network, VGG-16

CLC Number: