Real-time underwater target detection for AUV using side scan sonar images based on deep learning

Abstract

To overcome the shortcomings of traditional methods for underwater target detection in sonar images, a novel target detection algorithm based on deep learning is proposed for the real-time detection for an autonomous underwater vehicle equipped with a side scan sonar (SSS). First, the preprocessing of raw SSS images and data augmentation methods are proposed to improve the quality and quantity of the data set. Second, to improve the detection efficiency, a method based on threshold segmentation and pixel importance value is used to quickly determine whether there are suspected targets in the SSS images. Then, an MA-YOLOv7 network based on YOLOv7 with multi-scale information fusion and an attention mechanism is proposed to detect targets in the screened images, and a target localization method is proposed to obtain the location (latitude and longitude) of the target. Finally, the algorithm is verified through simulation experiments and field tests. The results demonstrate that the proposed algorithm achieves state-of-the-art performance and can be applied to real underwater tasks with a detection recall of 0.836 and time consumption of 0.355 s per image.