Orientation Robust Object Detection in Aerial Images Based on R-NMS

摘要

Object detection in aerial images is a challenging task which plays an important role in many fields, such as intelligent traffic management, fishery management and so on. Different from object detection in natural images, the orientation of objects in aerial images is arbitrary. The axis-aligned bounding box detection, which is always used in traditional object detection methods, will cover a lot of redundant information and deteriorate the detection results when it is used to locate the object in aerial images. Therefore, traditional object detection methods are no longer applicable for aerial images. In order to promote the object detection performance in aerial images, we propose a novel orientation robust object detection model based on rotated non-maximum suppression (R-NMS). In addition, we adjust the anchor setting according to the diversity shapes of the aerial objects to enhance the performance of the model. Our model is tested on the public DOTA dataset, and the mAP is 16.31% higher than the baseline, indicating that our method is very effective and competitive in the object detection of aerial image.