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 条
[1]   UAV remote sensing detection of tea leaf blight based on DDMA-YOLO [J].
Bao, Wenxia ;
Zhu, Ziqiang ;
Hu, Gensheng ;
Zhou, Xingen ;
Zhang, Dongyan ;
Yang, Xianjun .
COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2023, 205
[2]  
Bochkovskiy A., 2020, ARXIV, DOI [10.48550/ARXIV.2004.10934, 10.48550/arXiv.2004.10934]
[3]   A review on early wildfire detection from unmanned aerial vehicles using deep learning-based computer vision algorithms [J].
Bouguettaya, Abdelmalek ;
Zarzour, Hafed ;
Taberkit, Amine Mohammed ;
Kechida, Ahmed .
SIGNAL PROCESSING, 2022, 190
[4]  
Chen J., 2019, ARXIV191203619
[5]   Improved YOLOv3 Based on Attention Mechanism for Fast and Accurate Ship Detection in Optical Remote Sensing Images [J].
Chen, Liqiong ;
Shi, Wenxuan ;
Deng, Dexiang .
REMOTE SENSING, 2021, 13 (04) :1-18
[6]   Shallowing Deep Networks: Layer-wise Pruning based on Feature Representations [J].
Chen, Shi ;
Zhao, Qi .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2019, 41 (12) :3048-3056
[7]   Vehicle Detection in High-Resolution Aerial Images via Sparse Representation and Superpixels [J].
Chen, Ziyi ;
Wang, Cheng ;
Wen, Chenglu ;
Teng, Xiuhua ;
Chen, Yiping ;
Guan, Haiyan ;
Luo, Huan ;
Cao, Liujuan ;
Li, Jonathan .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2016, 54 (01) :103-116
[8]   Recent advances in efficient computation of deep convolutional neural networks [J].
Cheng, Jian ;
Wang, Pei-song ;
Li, Gang ;
Hu, Qing-hao ;
Lu, Han-qing .
FRONTIERS OF INFORMATION TECHNOLOGY & ELECTRONIC ENGINEERING, 2018, 19 (01) :64-77
[9]   Histograms of oriented gradients for human detection [J].
Dalal, N ;
Triggs, B .
2005 IEEE COMPUTER SOCIETY CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, VOL 1, PROCEEDINGS, 2005, :886-893
[10]   DSConv: Efficient Convolution Operator [J].
do Nascimento, Marcelo Gennari ;
Fawcett, Roger ;
Prisacariu, Victor Adrian .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :5147-5156