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采用更简单的模型和更少的回归参数即可实现先进的性能。