论文标题

单级旋转对象检测器通过太阳能电晕热图的两个点旋转

Single-stage Rotate Object Detector via Two Points with Solar Corona Heatmap

论文作者

Song, Beihang, Li, Jing, Xue, Shan, Chang, Jun, Wu, Jia, Wan, Jun, Liu, Tianpeng

论文摘要

面向对象检测是计算机视觉中的至关重要任务。当前的上下朝下检测方法通常直接检测整个对象,而不仅忽略了目标的真实方向,而且还不能完全利用关键语义信息,从而导致检测准确性降低。在这项研究中,我们通过带有太阳电晕热图(ROTP)的两个点开发了一个单级旋转对象检测器,以检测方向的物体。 ROTP预测对象的一部分,然后将它们汇总以形成整个图像。本文中,我们使用顶点,宽度和高度的中心点对象进行精心表示对象。具体而言,我们会回归两个热图,这些热图表征了每个对象的相对位置,从而提高了定位对象的准确性并避免了角度预测引起的偏差。为了纠正高斯热图对高估比目标的中央错误判断,我们设计了一种太阳电晕热图生成方法,以改善中央和非中央样品之间的感知差异。此外,我们预测了顶点的顶点,以连接属于同一目标的两个关键点。 HRSC 2016,UCASAOD和DOTA数据集的实验表明,我们的ROTP通过更简单的建模和较少的手动干预实现了最先进的性能。

Oriented object detection is a crucial task in computer vision. Current top-down oriented detection methods usually directly detect entire objects, and not only neglecting the authentic direction of targets, but also do not fully utilise the key semantic information, which causes a decrease in detection accuracy. In this study, we developed a single-stage rotating object detector via two points with a solar corona heatmap (ROTP) to detect oriented objects. The ROTP predicts parts of the object and then aggregates them to form a whole image. Herein, we meticulously represent an object in a random direction using the vertex, centre point with width, and height. Specifically, we regress two heatmaps that characterise the relative location of each object, which enhances the accuracy of locating objects and avoids deviations caused by angle predictions. To rectify the central misjudgement of the Gaussian heatmap on high-aspect ratio targets, we designed a solar corona heatmap generation method to improve the perception difference between the central and non-central samples. Additionally, we predicted the vertex relative to the direction of the centre point to connect two key points that belong to the same goal. Experiments on the HRSC 2016, UCASAOD, and DOTA datasets show that our ROTP achieves the most advanced performance with a simpler modelling and less manual intervention.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源