TS4Net: Two-stage sample selective strategy for rotating object detection

Jian Zhou,Kai Feng,Weixing Li,Jun Han,Feng Pan
DOI: https://doi.org/10.1016/j.neucom.2022.06.049
IF: 6
2022-08-28
Neurocomputing
Abstract:Rotating object detection has recently attracted increasing attention in aerial photographs, remote sensing images, etc. In this paper, we propose a rotating object detector TS4Net, which contains anchor refinement module (ARM) and two-stage sample selective strategy (TS4). The ARM can convert the preset horizontal anchors into high-quality rotated anchors through two-stage refinement, which also adopts a rotated Intersection-over-Union(IoU) prediction branch to improve localization accuracy in the second stage. TS4 module utilizes different constrained sample selective strategies to allocate positive and negative samples, which is adaptive to the regression task in different stages. Benefiting from the ARM and TS4, the TS4Net can achieve superior performance for rotating object detection solely with one preset horizontal anchor. Considering that most rotating object detection datasets mainly focus on the field of remote sensing and are shot in high-altitude scenes. We present a low-altitude drone-based dataset, named UAV-ROD, aiming to promote the research and development in rotating object detection and UAV applications. The UAV-ROD dataset can be used in rotating object detection, vehicle orientation recognition, and object counting tasks. Extensive experiments on the UAV-ROD dataset and four datasets demonstrate that our method achieves competitive performance against most state-of-the-art methods.
computer science, artificial intelligence
What problem does this paper attempt to address?