Real-Time Underwater Maritime Object Detection in Side-Scan Sonar Images Based on Transformer-YOLOv5

To overcome the shortcomings of the traditional manual detection of underwater targets in side-scan sonar (SSS) images, a real-time automatic target recognition (ATR) method is proposed in this paper. This method consists of image preprocessing, sampling, ATR by integration of the transformer module and YOLOv5s (that is, TR-YOLOv5s), and target localization. By considering the target-sparse and feature-barren characteristics of SSS images, a novel TR-YOLOv5s network and a down-sampling principle are put forward, and the attention mechanism is introduced in the method to meet the requirements of accuracy and efficiency for underwater target recognition. Experiments verified the proposed method achieved 85.6% mean average precision (mAP) and 87.8% macro-F-2 score, and brought 12.5% and 10.6% gains compared with the YOLOv5s network trained from scratch, and had the real-time recognition speed of about 0.068 s per image.

Automated summary

An real-time automatic target recognition method for underwater targets in SSS images was proposed in this paper, utilizing a combination of transformer module and YOLOv5s for target recognition and localization. The method, considering the characteristics of SSS images, introduced an attention mechanism, achieving superior performance in experiments.