SAR Ship Detection Based on YOLOv5 Using CBAM and BiFPN
Yue Guo, Shiqi Chen, Ronghui Zhan, Wei Wang, Jun Zhang
Abstract
In recent years, deep learning has made breakthroughs in the field of computer vision, the single-stage detection algorithm represented by You Only Look Once (YOLO) has achieved satisfying detection results in SAR ship target detection. For the multi-scale problem of SAR ship targets in complex scenes, we proposed an improved YOLOv5 detection method using Convolutional Block Attention Module (CBAM) and Bidirectional Feature Pyramid Network (BiFPN). The CBAM module and BiFPN are added in YOLOv5 so that it can fully learn the feature information of space and channel dimensions, and enhance information fusion transfer between multi-scale targets. Experiments on our dataset show that the proposed YOLOv5 algorithm achieves 92.8% Average Precision (AP), which gains a 1.9% improvement in AP compared to the standard YOLOv5 algorithm in SAR ship target detection. The problem of missed detection of multi-scale targets is well solved.