As the rapid development of depth learning, object detection in aviatic remote sensing images has become increasingly popular in recent years. Most of the current Anchor Free detectors based on key point detection sampling directly regression and classification features, with the design of object loss function based on the horizontal bounding box. It is more challenging for complex and diverse aviatic remote sensing object. In this paper, we propose an Anchor Free aviatic remote sensing object detector (BWP-Det) to detect rotating and multi-scale object. Specifically, we design a interactive double-branch(IDB) up-sampling network, in which one branch gradually up-sampling is used for the prediction of Heatmap, and the other branch is used for the regression of boundary box parameters. We improve a weighted multi-scale convolution (WmConv) in order to highlight the difference between foreground and background. We extracted Pixel level attention features from the middle layer to guide the two branches to pay attention to effective object information in the sampling process. Finally, referring to the calculation idea of horizontal IoU, we design a rotating IoU based on the split polar coordinate plane, namely JIoU, which is expressed as the intersection ratio following discretization of the inner ellipse of the rotating bounding box, to solve the correlation between angle and side length in the regression process of the rotating bounding box. Ultimately, BWP-Det, our experiments on DOTA, UCAS-AOD and NWPU VHR-10 datasets show, achieves advanced performance with simpler models and fewer regression parameters.
翻译:随着深度学习的快速发展,航空遥感图像中的目标检测在近年来变得越来越流行。当前大多数基于无锚点的检测器是基于关键点检测采样直接回归和分类特征的,并基于水平边界框设计了对象损失函数。这对于复杂和多样的航空遥感对象来说更具挑战性。本文中,我们提出了一种基于离散极坐标方程的无锚点航空遥感目标检测器(BWP-Det),以检测旋转和多尺度对象。具体来说,我们设计了一个交互式双支(IDB)上采样网络,其中一个逐渐上采样的分支用于预测热图,另一个分支用于边界框参数的回归。我们改进了加权多尺度卷积(WmConv),以凸显前景和背景之间的差异。我们从中间层提取像素级注意特征,以指导两个分支在采样过程中关注有效的对象信息。最后,参考水平IoU的计算思想,我们设计了一个基于极坐标平面的旋转IoU,即JIoU,它表示离散化旋转边界框内椭圆的交集比率,以解决旋转边界框回归过程中角度和边长之间的相关性。最终,我们在DOTA,UCAS-AOD和NWPU VHR-10数据集上的实验表明,BWP-Det使用更简单的模型和更少的回归参数实现了先进的性能。