基于多头自注意力的复杂背景船舶检测算法

于楠晶,范晓飚,邓天民,冒国韬

Ship detection algorithm in complex backgrounds via multi-head self-attention

Nan-jing YU,Xiao-biao FAN,Tian-min DENG,Guo-tao MAO

表 1 算法性能的消融实验结果对比

Tab.1 Comparison of ablation experimental results of algorithm performance

算法 pma/% F1 Fps/(帧·s−1) M/MB FLOPS/109 Par YOLOv5 96.30 0.86 26 13.7 16.4 7 277 027 YOLOv5+简化的FPN 96.73 0.91 25 15.7 17.8 8 145 079 YOLOv5+PARM 96.51 0.93 27 13.1 16.1 6 760 099 MHSA-YOLO 97.59 0.94 25 15.1 17.5 7 828 151