首页|期刊导航|电子学报|面向遥感目标检测的无锚框Transformer算法

面向遥感目标检测的无锚框Transformer算法OACSCDCSTPCD

Anchor-Free Transformer Algorithm for Aerial Remote Sensing Target Detection

中文摘要英文摘要

遥感图像目标具有多方向排布、小且密集等特性,使基于深度学习的旋转目标检测算法存在检测精度不佳的问题.针对这一问题,本文提出了一种面向遥感目标检测的无锚框Transformer算法.首先,采用层次化Trans-former采集不同分辨率的特征信息以扩大特征信息的采集范围.其次,构建一种新的前馈网络(Spacial-FeedForward Neural network,SFFN).SFFN将3´3深度可分离卷积的局部空间特性和多层感知机(MultiLayer Perceptron,MLP)的全局通道特性融合在一起,以解决前馈网络(Feed Forward Neural network,FFN)在局部空间建模上的不足.最后,基于SFFN架构搭建了无锚框检测器,将预测框回归问题分为水平框与旋转框,缓解了旋转框的损失不连续性问题.在DO-TA数据集上的测试结果表明,此方法的平均精度达到了75.83%,同时在NWPU VHR-10数据集上5类小目标检测结果达到了92.47%,在遥感目标检测精度上更具竞争力.

Aerial remote sensing image targets have the characteristics of multi-directional arrangement,small,and dense.The rotating target detection algorithm based on deep learning has the problem of poor detection accuracy.To solve this problem,the article proposes a novel anchor-free Transformer algorithm for aerial remote sensing target detection.Firstly,hierarchical Transformer is used to collect feature information of different resolutions to improve the range of fea-ture information collection.Secondly,a new feedforward network(Spacial-FeedForward Neural network,SFFN)is con-structed.SFFN combines the local space characteristics of 3×3 depth separable convolution with the global channel charac-teristics of multi-layer perceptron(MLP)to solve the shortcomings of feed forward neural network(FFN)in local space modeling.Finally,an anchor-free detector is built based on SFFN architecture,and the regression problem of prediction frame is divided into horizontal frame and rotating frame,which alleviates the loss discontinuity problem of rotating frame.The test results on DOTA dataset show that the average accuracy of this method has reached 75.83%,respectively,while achieving 92.47%of 5 small targets on NWPU VHR-10 dataset,which is more competitive in remote sensing target detec-tion accuracy.

喻九阳;胡天豪;戴耀南;张德安;夏文凤

武汉工程大学机电工程学院湖北省绿色化工装备工程技术研究中心,湖北武汉 430205武汉工程大学机电工程学院湖北省绿色化工装备工程技术研究中心,湖北武汉 430205武汉工程大学机电工程学院湖北省绿色化工装备工程技术研究中心,湖北武汉 430205武汉工程大学机电工程学院湖北省绿色化工装备工程技术研究中心,湖北武汉 430205武汉工程大学机电工程学院湖北省绿色化工装备工程技术研究中心,湖北武汉 430205

计算机与自动化

遥感图像目标检测Transformer算法无锚框检测器

remote sensing imagetarget detectionTransformer algorithmanchor-free detector

《电子学报》 2023 (11)

3238-3247,10

湖北省重点研发计划(No.2020BAB030)湖北省自然科学基金(No.2023AFC010)Hubei Provincial Key R&D Project(No.2020BAB030)Natural Science Foundation of Hubei Province(No.2023AFC010)

10.12263/DZXB.20220612

评论