A Feature Pyramid Based Multi-stage Framework for Object Detection in Low-altitude UAV Images

Abstract

This paper proposes a high-performance framework for accurate multi-stage object detection in low-altitude based UAV images. The proposed system employs a cascade style architecture with increasing thresholds for achieving accurate detection. The framework makes use of highly efficient Feature Pyramid Networks (FPNs) to detect objects of small sizes, and various scales which are the main challenge in low-altitude aerial images. FPNs aim to resolve scale variation problems in object detection by combining features of multiple levels. The experiments have been performed on a complex low-altitude aerial dataset VisDrone which has multiple categories of classes. The FPN-Cascade detector has been supported by slicing the data horizontally and vertically that resulted in an advancement of 8% mAP when compared with the base detector. The experiments compare the FPN-Cascade performance on the standard as well as augmented VisDrone dataset. A concrete methodology about the training process, hyperparameter tuning, and performance evaluation methods for Cascade RCNN on the VisDrone dataset is highlighted. The proposed framework achieves state of the art 30.04% mAP value on the VisDrone dataset.