IET Image Processing (Aug 2024)
Maritime distress target detection algorithm based on YOLOv5s‐EFOE network
Abstract
Abstract Traditional maritime search and rescue methods mainly rely on manual operation, which takes a long time to identify and results in poor search and rescue results. This paper applies computer vision technology to the field of maritime distress object detection and proposes an improved object detection algorithm YOLOv5s‐EFOE based on the YOLOv5s algorithm. Firstly, the authors change the detection head of the YOLOv5s algorithm to use a mixed channel strategy to build a more efficient decoupling head, reducing the number of 3 × 3 convolutional layers in the middle of the decoupling head to only one. The width of the head is scaled by the width multipliers of Backbone and Neck. Secondly, in conjunction with the SimOTA matching algorithm, the positive samples of the target to be tested are dynamically allocated to improve the recognition ability of different targets. Finally, considering the low pixel value of the maritime distress target from the perspective of unmanned aerial vehicles (UAV), the loss function CIoU in the original YOLOv5s is changed to EIoU. EIoU not only considers the distance and aspect ratio of the centre point, but also considers the true difference between the width and height of the prediction box and the real box, which improves the prediction accuracy of the anchor box. Experiments are conducted on a subset of the public dataset SeaDronesSee. The Recall of the YOLOv5s‐EFOE model proposed in this paper reached 75.9%, mAP0.5 reached 79.9%, and mAP0.5−0.95 reached 44.5%. These indicators are superior to the original YOLOv5s model, YOLOv7 series models, and YOLOv8 series models. Compared with the original model, the YOLOv5s‐EFOE model has increased the Recall by 5.4%, mAP0.5 by 5.6%, and mAP0.5−0.95 by 4.6%, improving the difficult to detect and missed detection situations. This model can be deployed on UAVs and can effectively identify maritime distress target, achieving the purpose of search and rescue.
Keywords