Enhancing Small Object Detection with YOLO: A Novel Framework for Improved Accuracy and Efficiency

TL;DR

This paper introduces a novel YOLO-based framework with architectural enhancements and improved cropping techniques to significantly boost small object detection accuracy in aerial images, outperforming existing methods like SAHI and CZDet.

Contribution

The paper presents a new model that combines advanced feature extraction, attention mechanisms, and a specialized detection head to improve small object detection accuracy in large-scale aerial imagery.

Findings

Achieved 61.2% mAP on VisDrone2019 dataset, surpassing YOLOv5L baseline.

Outperformed SAHI and CZDet frameworks in accuracy.

Demonstrated significant improvements in small object detection performance.

Abstract

This paper investigates and develops methods for detecting small objects in large-scale aerial images. Current approaches for detecting small objects in aerial images often involve image cropping and modifications to detector network architectures. Techniques such as sliding window cropping and architectural enhancements, including higher-resolution feature maps and attention mechanisms, are commonly employed. Given the growing importance of aerial imagery in various critical and industrial applications, the need for robust frameworks for small object detection becomes imperative. To address this need, we adopted the base SW-YOLO approach to enhance speed and accuracy in small object detection by refining cropping dimensions and overlap in sliding window usage and subsequently enhanced it through architectural modifications. we propose a novel model by modifying the base model architecture, including advanced feature extraction modules in the neck for feature map enhancement, integrating CBAM in the backbone to preserve spatial and channel information, and introducing a new head to boost small object detection accuracy. Finally, we compared our method with SAHI, one of the most powerful frameworks for processing large-scale images, and CZDet, which is also based on image cropping, achieving significant improvements in accuracy. The proposed model achieves significant accuracy gains on the VisDrone2019 dataset, outperforming baseline YOLOv5L detection by a substantial margin. Specifically, the final proposed model elevates the mAP .5.5 accuracy on the VisDrone2019 dataset from the base accuracy of 35.5 achieved by the YOLOv5L detector to 61.2. Notably, the accuracy of CZDet, which is another classic method applied to this dataset, is 58.36. This research demonstrates a significant improvement, achieving an increase in accuracy from 35.5 to 61.2.