Global maritime transport is booming, and with it the need to spot vessels even at night or in poor visibility. Scientists have tried to improve infrared ship detection, but traditional methods struggle with tiny targets and cluttered backgrounds.
In this study, researchers embedded the CBAM (Convolutional Block Attention Module) into the YOLOv5 network, a popular fast object detector. They built three versions: YOLOv5nβH (attention in the head), YOLOv5nβN (attention in the neck) and YOLOv5nβHN (attention in both head and neck).
Tests on a public infrared ship dataset yielded impressive results. The YOLOv5nβHN model achieved an mAP@0.5 of 86.83β―%, especially boosting detection of mediumβ and largeβsize vessels. YOLOv5nβN excelled at spotting small boats. Moreover, attention made the model more robust against background noise, reducing false alarms.
These findings suggest that strategically placing attention mechanisms can make maritime surveillance systems lighter, faster and more reliable, paving the way for new safety and environmental protection applications.
