Ship visible images detection based on computer vision plays an important role in the field of intelligent ship. To increase the model speed, accuracy, and reduce the parameters of the model to facilitate the deployment on hardware devices in practical applications, this study proposed a new model named STD-Yolov5. Firstly, the attention mechanism module of ECA was embedded in backbone to enhance the feature extraction capability of the network. Secondly, GAFPN was designed to reduce the parameters and GFLOPs. Thirdly, to solve the problem of ship-type false detection and missing detection, this paper presented a new receptive field amplification module named GSPP. Finally, replaced the GIoU bounding box regression loss function with a simpler generalisation of α-GIoU to improve the accuracy of the model. Compared to Yolov5, the mAP@.5:.95 of STD-Yolov5 increased by 1.2%, the parameters decreased by 24.85%, and the GFLOPs decreased by 14.46%.
STD-Yolov5: a ship-type detection model based on improved Yolov5
SHIPS AND OFFSHORE STRUCTURES
Y. NING ET AL.
Ships and Offshore Structures ; 19 , 1 ; 66-75
02.01.2024
10 pages
Aufsatz (Zeitschrift)
Elektronische Ressource
Englisch
Traffic Sign Detection Based on Improved YOLOv5 Model
Springer Verlag | 2024
|Shipping Risk Assessment Ship Inspection Method Based on Improved YOLOv5
Springer Verlag | 2024
|