YOLO-ME: an enhanced lightweight YOLOv7 Tiny model for efficient object detection in aerial imagery
Document Type
Article
Publication Date
4-1-2025
Abstract
Vehicle detection in aerial images presents several challenges, including small object sizes, complex backgrounds, varied orientations, and scale variability, all influenced by changing illumination, weather conditions, and high image resolution. Current one-stage detection models have achieved excellent accuracy; however, their complex structures are unsuitable for real-time aerial object detection on embedded devices. To address this issue, this paper proposes the YOLO-ME model, developed based on the YOLOv7 Tiny architecture. The proposed model incorporates three major modifications, including adopting a novel SEELAN module in the backbone, the MELAN module in the neck section, and the adoption of Swish activation functions in most convolutional layers. Experimental results evaluated on the VisDrone dataset show that the YOLO-ME model surpasses other lightweight YOLO-based models, attaining an mAP of 24.82%. Additionally, it features the smallest model size and BFLOPs value at 18.8 MB and 4.68 BFLOPs, respectively. The proposed model demonstrates a detection speed comparable to the YOLOv7 Tiny model. The results indicate that the YOLO-ME model has great potential for real-time aerial object detection on embedded systems. Our code is available at: https://github.com/hanifjunos/YOLO-ME.
Keywords
Aerial object detection, Deep learning, YOLO, Modified ELANS
Divisions
sch_ecs
Funders
Universiti Sains Malaysia [Grant no. R501-LR-RND002-0000000130-0000]
Publication Title
Signal, Image and Video Processing
Volume
19
Issue
4
Publisher
Springer Nature
Publisher Location
236 GRAYS INN RD, 6TH FLOOR, LONDON WC1X 8HL, ENGLAND