基于UPCBAM-RYOLO V5的光学遥感舰船小目标检测
Optical Remote Sensing Ship Small Target Detection Based on UPCBAM-RYOLO V5
杨笑天 1鱼昕 1刘铭 2王梁 1谭金林 1吴意1
作者信息
- 1. 西安空间无线电技术研究所,西安 710100;陕西航天技术应用研究院有限公司系统总体部,西安 710100
- 2. 长春工业大学数学与统计学院,长春 130012
- 折叠
摘要
针对光学遥感数据中舰船小目标数量占比大,长宽比大且多个目标紧密排列难以检测的问题,提出一种基于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模型在光学遥感舰船小目标检测方面具有一定的研究意义.
Abstract
In order to solve the problems of large proportion of small targets in optical remote sensing data,the aspect ratio is large,and multiple targets are closely arranged and difficult to detect,we present an optical remote sensing small ship target detection algorithm based on the YOLO V5(You Only Look Once V5),UPCBAM-RYOLO V5(Upsampling Convolutional Attention Block Module-RYOLO V5)algorithm.An up-sampling attention mechanism module is designed to enhance the feature extraction ability of small size targets.The rotation angle loss is introduced into the frame regression formula to improve the algorithm's perception ability of the ship's appearance and direction.Based on the experiment of small ship target datasets composed of GF1 and GF2,the results show that the UPCBAM-RYOLO V5 algorithm model improves the positioning accuracy and classification accuracy of small ship target detection.The P value,R value,and MAP(Mean Average Precision)value reach 93％,94％,and 95％respectively,which are 3％,7％,and 7％higher than the original YOLO V5 model.In the upsampled attention-mechanism module added location ablation experiment to the network,the results show that compared with the addition of UPCBAM in Backbone and Prediction,the addition of UPCBAM in Neck has the greatest influence on the detection of small target ships in remote sensing images.The performance is the best,with P value,R value,and MAP value increased by 5％,4％,and 2％,respectively.The UPCBAM-RYOLO V5 model is proved to have a certain research significance in optical remote sensing ship small target detection.
关键词
光学遥感数据/舰船小目标/UPCBAM-RYOLO/V5算法/上采样注意力机制/旋转角度Key words
optical remote sensing data/small ship targets/upsampling convolutional attention block module-R you only look once V5(UPCBAM-RYOLOV5)algrithom/upsampling convolutional attention block module/rotation angle引用本文复制引用
出版年
2024
国家科技期刊平台
NSTL
万方数据