首页 | 本学科首页   官方微博 | 高级检索  
     

基于改进FCOS网络的光学遥感影像目标检测算法
引用本文:杨玉春,王腾军,任会涛,杨耘.基于改进FCOS网络的光学遥感影像目标检测算法[J].计算机测量与控制,2022,30(5):6-12.
作者姓名:杨玉春  王腾军  任会涛  杨耘
作者单位:长安大学地质工程与测绘学院,西安 710054
基金项目:陕西省自然科学基金(2022JM-163)
摘    要:针对当前遥感影像背景复杂、目标尺度小等情况导致的检测精度偏低的问题,基于FCOS网络提出了一种结合位置注意力和感受野增强的遥感影像目标检测算法PARF-FCOS;该算法构造了一种位置注意力模块,并利用该模块对特征提取网络进行改进,增强网络对目标信息的提取能力;在特征融合阶段使用感受野模块(RFB, receptive field block)增强浅层特征图,利用目标上下文信息进行辅助判断,提升网络对小尺度目标的检测能力;在训练过程中,引入距离交并比损失(DIoU loss,distance intersection over union loss)进行边界框回归,通过优化目标框与预测框中心点之间的距离,使回归过程更加平稳和准确;在公开数据集DIOR上评估了PARF-FCOS目标检测算法,实验结果表明,相较于原始FCOS,算法的平均精确度均值提高了4.3%,达到70.4%,检测速度达到23.2FPS。

关 键 词:遥感影像处理  目标检测  卷积神经网络  注意力机制  无锚框
收稿时间:2022/3/3 0:00:00
修稿时间:2022/3/24 0:00:00

Object Detection Algorithm of Optical Remote Sensing Imagery Based on An Improved FCOS Network

YANG Yuchun,WANG Tengjun,REN Huitao,YANG Yun.

Object Detection Algorithm of Optical Remote Sensing Imagery Based on An Improved FCOS Network[J].Computer Measurement & Control,2022,30(5):6-12.

Authors:YANG Yuchun  WANG Tengjun  REN Huitao  YANG Yun
Abstract:Aiming at the problem of low detection accuracy caused by complex background and small objects in remote sensing imageries, a remote sensing imagery object detection algorithm PARF-FCOS based on FCOS network is proposed. The algorithm constructs a position attention module, and uses the module to reconstruct the feature extraction network to enhance the ability of the network to extract target information; In the feature fusion stage, RFB(receptive field block) is used to enhance the shallow feature map, and the target context information is used for auxiliary judgment to improve the detection ability of the network for small-scale objects; During training, DIOU loss(distance intersection over union loss) is introduced for boundary box regression. By optimizing the distance between the center point of the target box and the prediction box, the regression process is more stable and accurate. Experiments are carried out with public dataset DIOR. Compared with the original FCOS, the mean Average Precision of the algorithm is improved by 4.3%, up to 70.4%, and the detection speed is 23.2 FPS.
Keywords:remote sensing imagery processing  object detection  convolutional neural network  attention mechanism  anchor-free
本文献已被 万方数据 等数据库收录!
点击此处可从《计算机测量与控制》浏览原始摘要信息
点击此处可从《计算机测量与控制》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号