APS-Net: An Adaptive Point Set Network for Optical Remote-Sensing Object Detection

Abstract

Oriented object detection in optical remote-sensing images has been a challenging task due to arbitrary orientations and densely packed distribution of objects. Specifically, most existing methods lack adaptivity when regressing objects with different shapes and orientations. Although the point set representation is relatively flexible, the initial distribution of the point set is fixed in advance. In addition, some models based on the point set cannot get high location precision of points, affecting the bounding box generation. In this letter, we propose an Adaptive Point Set Network (APS-Net) for optical remote-sensing object detection, including three improvements. First, we propose the initial distribution learner (IDL) to learn the optimal initial aspect ratio, which helps the point set fit the object’s shape well. Second, we design the uncertainty measurement module (UMM), which considers the uncertainty of point location to improve location precision. Third, we introduce the local outlier factor (LOF) in the loss to punish outlier points more reasonably. Extensive experiments demonstrate that our proposed model achieves state-of-the-art performance on three commonly used datasets (i.e., DOTA-v1.0, UCAS-AOD, and HRSC2016) in the remote-sensing field.