遥感场景下基于关系挖掘的旋转目标检测算法研究

Research on Rotating Target Detection Algorithm Based on Relation Mining in Remote Sensing Scene

与常规场景相比,遥感场景目标检测任务存在图像尺寸大、小目标数量多、检测框有旋转角等难点,这些难点也使得遥感图像中物体间有更多的关系可挖掘。为提升遥感场景下对旋转目标的检测效果,通过添加关系挖掘模块对旋转目标检测算法(Oriented R-CNN for Object Detection, ORCN)进行优化。关系挖掘模块利用动态图神经网络、交叉注意力机制使候选区域的特征、形状信息进行有效交互,丰富候选区域特征的上下文语义。实验结果表明,添加关系挖掘模块后模型在遥感数据集上的DOTA表现提升1.53%,明显优于原检测算法。

Compared with the conventional scene,the target detection task in remote sensing scene has difficulties such as large image size,large number of small targets,and detection frame with rotation angle.These difficulties also make more relationships between objects in remote sensing images can be mined.In order to improve the detection effect of rotating targets in remote sensing scenes,the Oriented R-CNN for Object Detection(ORCN)algorithm is optimized by adding a relationship mining module.The relationship mining module uses dynamic graph neural network and cross-attention mechanism to effectively interact the features and shape information of candidate regions,and enrich the context semantics of the features of candidate regions.The experimental results show that the DOTA performance of the model on the remote sensing data set is improved by 1.53%after adding the relationship mining module,which is significantly better than the original detection algorithm.