Unsupervised Cluster Guided Object Detection in Aerial Images

Object detection from high-resolution aerial images have received increasing attention during the last few years. It is a common practice to downsize images before feeding them into a network. In real life, there are lots of scenes where many objects gather together in certain areas, such as crossroads, parking lots, and playgrounds. The downsizing operation significantly limits the detection ability in these scenes. In this paper, we proposed an unsupervised cluster guided detection framework (UCGNet) to address these issues by guiding the detector focus on the object-densely distributed area. In particular, a local location module is first applied to predict a binary map presenting how objects distribute in terms of the pixel of the map. Then, an unsupervised cluster method is used to produce dense regions. Each adjusted dense region is fed into the detector for object detection. Finally, a global merge module generates the final predict results. Experiments were conducted on two popular aerial image datasets including VisDrone2019 and UAVDT. In both datasets, our proposed method outperforms the existing baseline methods with achieving 32.8% mAP and 19.1% mAP, respectively. The code is available at https://github.com/jason-su/UCGNET.