EL-Net: An efficient and lightweight optimized network for object detection in remote sensing images

被引:9
作者
Dong, Chao [1 ]
Jiang, Xiangkui [1 ]
Hu, Yihui [1 ]
Du, Yaoyao [1 ]
Pan, Libing [1 ]
机构
[1] Xian Univ Posts & Telecommun, Sch Automat, Xian 710121, Peoples R China
关键词
UAV; Object detection; Small objects; Lightweight design; HAAR-LIKE FEATURES;
D O I
10.1016/j.eswa.2024.124661
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Object detection in Unmanned Aerial Vehicles (UAV) optical remote sensing imagery presents a formidable challenge in computer vision due to the minuscule size of targets, which occupy fewer pixels and provide limited feature information, complicating accurate recognition and classification. Furthermore, the overlapping of dense targets exacerbates the difficulty of precise classification and localization. Meanwhile, classical detection networks often struggle to balance recognition accuracy with model complexity. Addressing these issues, this paper introduces EL-Net, an efficient and lightweight network model based on improvements to the YOLOv7-tiny architecture. First, the network structure is streamlined through a lightweight design that maintains performance while reducing complexity. Additionally, a feature perception enhancement module (FPEM) using attention mechanisms and dilated convolution significantly improves the model's capability to extract key features from complex backgrounds. Finally, the optimized network structure is compressed by a structured pruning algorithm. EL-Net was evaluated in challenging scenarios on the VisDrone2019 dataset, where it achieved a mean Average Precision (mAP) of 38.7%, demonstrating high detection accuracy at minimal model complexity. Meanwhile, evaluation of the UA-DETRAC dataset has demonstrated the model's remarkable generalization capacity. The outcomes suggest that EL-Net effectively balances accuracy and efficiency, making it ideal for deployment on resource-limited mobile edge devices while offering an innovative approach to object detection in UAV imagery.
引用
收藏
页数:15
相关论文
共 59 条
[41]  
Redmon J, 2018, Arxiv, DOI arXiv:1804.02767
[42]   Drones in agriculture: A review and bibliometric analysis [J].
Rejeb, Abderahman ;
Abdollahi, Alireza ;
Rejeb, Karim ;
Treiblmaier, Horst .
COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2022, 198
[43]   Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks [J].
Ren, Shaoqing ;
He, Kaiming ;
Girshick, Ross ;
Sun, Jian .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2017, 39 (06) :1137-1149
[44]  
Selvaraju RR, 2020, INT J COMPUT VISION, V128, P336, DOI [10.1109/ICCV.2017.74, 10.1007/s11263-019-01228-7]
[45]  
Shazeer N., 2017, arXiv
[46]   SAR Target Recognition via Supervised Discriminative Dictionary Learning and Sparse Representation of the SAR-HOG Feature [J].
Song, Shengli ;
Xu, Bin ;
Yang, Jian .
REMOTE SENSING, 2016, 8 (08)
[47]   YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors [J].
Wang, Chien-Yao ;
Bochkovskiy, Alexey ;
Liao, Hong-Yuan Mark .
2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR, 2023, :7464-7475
[48]   CSPNet: A New Backbone that can Enhance Learning Capability of CNN [J].
Wang, Chien-Yao ;
Liao, Hong-Yuan Mark ;
Wu, Yueh-Hua ;
Chen, Ping-Yang ;
Hsieh, Jun-Wei ;
Yeh, I-Hau .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW 2020), 2020, :1571-1580
[49]   Remote sensing image super-resolution and object detection: Benchmark and state of the art [J].
Wang, Yi ;
Bashir, Syed Muhammad Arsalan ;
Khan, Mahrukh ;
Ullah, Qudrat ;
Wang, Rui ;
Song, Yilin ;
Guo, Zhe ;
Niu, Yilong .
EXPERT SYSTEMS WITH APPLICATIONS, 2022, 197
[50]   UA-DETRAC: A new benchmark and protocol for multi-object detection and tracking [J].
Wen, Longyin ;
Du, Dawei ;
Cai, Zhaowei ;
Lei, Zhen ;
Chang, Ming-Ching ;
Qi, Honggang ;
Lim, Jongwoo ;
Yang, Ming-Hsuan ;
Lyu, Siwei .
COMPUTER VISION AND IMAGE UNDERSTANDING, 2020, 193