基于UPCBAM-RYOLO V5的光学遥感舰船小目标检测

杨笑天; 鱼昕; 刘铭; 王梁; 谭金林; 吴意

doi:10.19292/j.cnki.jdxxp.20240305.001

吉林大学学报(信息科学版) ›› 2024, Vol. 42 ›› Issue (06) : 1048 -1057. DOI: 10.19292/j.cnki.jdxxp.20240305.001

基于UPCBAM-RYOLO V5的光学遥感舰船小目标检测

杨笑天, 鱼昕, 刘铭, 王梁, 谭金林, 吴意

作者信息 +

Author information +

文章历史 +

摘要

针对光学遥感数据中舰船小目标数量占比大，长宽比大且多个目标紧密排列难以检测的问题，提出一种基于YOLO V5(You Only Look Once V5)的光学遥感舰船小目标检测算法——UPCBAM-RYOLO V5(Upsampling Convolutional Attention Block Module-RYOLO V5)算法。该算法设计了一种上采样注意力机制模块，增强了网络对小尺寸目标特征提取能力，并在边框回归公式中引入旋转角度损失，提高了算法对舰船外观和方向的感知能力。基于GF1、GF2组成的舰船小目标数据集实验，结果表明，UPCBAM-RYOLO V5算法模型提升了舰船小目标检测的定位精度和分类精度，其中P、R、 MAP(Mean Average Precision)值分别达到93%、94%和95%,较传统YOLO V5模型分别提高3%、7%和7%。对网络中上采样注意力机制模块添加位置的消融实验结果表明，相较于在Backbone和Prediction处加入UPCBAM,在Neck处加入UPCBAM对遥感影像舰船小目标的检测影响最大，性能最优，P、R和MAP值分别提高了5%、4%和2%。从而验证了UPCBAM-RYOLO V5模型在光学遥感舰船小目标检测方面具有一定的研究意义。