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 条
[21]   Coordinate Attention for Efficient Mobile Network Design [J].
Hou, Qibin ;
Zhou, Daquan ;
Feng, Jiashi .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :13708-13717
[22]  
Howard AG., 2017, ARXIV
[23]   SS R-CNN: Self-Supervised Learning Improving Mask R-CNN for Ship Detection in Remote Sensing Images [J].
Jian, Ling ;
Pu, Zhiqi ;
Zhu, Lili ;
Yao, Tiancan ;
Liang, Xijun .
REMOTE SENSING, 2022, 14 (17)
[24]   Deep learning based object detection for resource constrained devices: Systematic review, future trends and challenges ahead [J].
Kamath, Vidya ;
Renuka, A. .
NEUROCOMPUTING, 2023, 531 :34-60
[25]   ImageNet Classification with Deep Convolutional Neural Networks [J].
Krizhevsky, Alex ;
Sutskever, Ilya ;
Hinton, Geoffrey E. .
COMMUNICATIONS OF THE ACM, 2017, 60 (06) :84-90
[26]   Fast detection and location of longan fruits using UAV images [J].
Li, Denghui ;
Sun, Xiaoxuan ;
Elkhouchlaa, Hamza ;
Jia, Yuhang ;
Yao, Zhongwei ;
Lin, Peiyi ;
Li, Jun ;
Lu, Huazhong .
COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2021, 190 (190)
[27]  
Li H., 2022, arXiv
[28]   Scale-Aware Trident Networks for Object Detection [J].
Li, Yanghao ;
Chen, Yuntao ;
Wang, Naiyan ;
Zhang, Zhaoxiang .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :6053-6062
[29]   Deep Learning-Based Object Detection Techniques for Remote Sensing Images: A Survey [J].
Li, Zheng ;
Wang, Yongcheng ;
Zhang, Ning ;
Zhang, Yuxi ;
Zhao, Zhikang ;
Xu, Dongdong ;
Ben, Guangli ;
Gao, Yunxiao .
REMOTE SENSING, 2022, 14 (10)
[30]  
Lienhart R, 2002, IEEE IMAGE PROC, P900