A Lightweight YOLOv5-MNE Algorithm for SAR Ship Detection

Sensors (Basel). 2022 Sep 19;22(18):7088. doi: 10.3390/s22187088.

Abstract

Unlike optical satellites, synthetic aperture radar (SAR) satellites can operate all day and in all weather conditions, so they have a broad range of applications in the field of ocean monitoring. The ship targets' contour information from SAR images is often unclear, and the background is complicated due to the influence of sea clutter and proximity to land, leading to the accuracy problem of ship monitoring. Compared with traditional methods, deep learning has powerful data processing ability and feature extraction ability, but its complex model and calculations lead to a certain degree of difficulty. To solve this problem, we propose a lightweight YOLOV5-MNE, which significantly improves the training speed and reduces the running memory and number of model parameters and maintains a certain accuracy on a lager dataset. By redesigning the MNEBlock module and using CBR standard convolution to reduce computation, we integrated the CA (coordinate attention) mechanism to ensure better detection performance. We achieved 94.7% precision, a 2.2 M model size, and a 0.91 M parameter quantity on the SSDD dataset.

Keywords: YOLOv5; deep learning; lightweight model; ship detection.