Improving Training Instance Quality in Aerial Image Object Detection With a Sampling-Balance-Based Multistage Network

Abstract

Object detection, aiming to recognize and locate objects of interest in aerial images, has historically played a significant role in the remote sensing community. Following remarkable improvements in Earth observation technologies, high-resolution remote sensing (HRRS) images with a bird’s eye view perspective have revealed many categories of objects with sufficient variations in appearance and on complex backgrounds that make HRRS object detection an active but challenging task. The selection of positive samples and negative training instances is an essential factor in influencing detectors’ performance. Related studies have found that many low-quality negative samples in the detectors’ training process have caused training instability and low detection accuracy. In this work, a novel sampling-balance-based multistage network (SB-MSN) is presented to adaptively mine high-quality positive and negative instances for training an accurate detector. It has a series of components to ensure the selection and generation of high-quality examples for training an accurate detector, including a multiscale information retention module, an intersection over union balance sampling strategy, a balance L1 loss, and a multistage network. The proposed detector has been evaluated on three representative HRRS data sets. The extensive experimental results show that our detector can solve the problem of low-quality samples and significantly improve the detection performance of the mAP by 1.4% with the NWPU VHR-10 data set, 3.5% with the high-resolution remote sensing detection (HRRSD) data set, and 4.2% with the detection in the optical remote (DIOR) data set. <xref ref-type="fn" rid="fn1" xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">¹</xref>