LI-YOLO: An Object Detection Algorithm for UAV Aerial Images in Low-Illumination Scenes
With the development of unmanned aerial vehicle (UAV) technology, deep learning is becoming more and more widely used in object detection in UAV aerial images; however, detecting and identifying small objects in low-illumination scenes is still a major challenge. Aiming at the problem of low brightn...
Saved in:
| Main Authors: | , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
MDPI AG
2024-11-01
|
| Series: | Drones |
| Subjects: | |
| Online Access: | https://www.mdpi.com/2504-446X/8/11/653 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1846153737597353984 |
|---|---|
| author | Songwen Liu Hao He Zhichao Zhang Yatong Zhou |
| author_facet | Songwen Liu Hao He Zhichao Zhang Yatong Zhou |
| author_sort | Songwen Liu |
| collection | DOAJ |
| description | With the development of unmanned aerial vehicle (UAV) technology, deep learning is becoming more and more widely used in object detection in UAV aerial images; however, detecting and identifying small objects in low-illumination scenes is still a major challenge. Aiming at the problem of low brightness, high noise, and obscure details of low-illumination images, an object detection algorithm, LI-YOLO (Low-Illumination You Only Look Once), for UAV aerial images in low-illumination scenes is proposed. Specifically, in the feature extraction section, this paper proposes a feature enhancement block (FEB) to realize global receptive field and context information learning through lightweight operations and embeds it into the C2f module at the end of the backbone network to alleviate the problems of high noise and detail blur caused by low illumination with very few parameter costs. In the feature fusion part, aiming to improve the detection performance for small objects in UAV aerial images, a shallow feature fusion network and a small object detection head are added. In addition, the adaptive spatial feature fusion structure (ASFF) is also introduced, which adaptively fuses information from different levels of feature maps by optimizing the feature fusion strategy so that the network can more accurately identify and locate objects of various scales. The experimental results show that the mAP50 of LI-YOLO reaches 76.6% on the DroneVehicle dataset and 90.8% on the LLVIP dataset. Compared with other current algorithms, LI-YOLO improves the mAP 50 by 3.1% on the DroneVehicle dataset and 6.9% on the LLVIP dataset. Experimental results show that the proposed algorithm can effectively improve object detection performance in low-illumination scenes. |
| format | Article |
| id | doaj-art-8fec63294bbc45fc9cb09db80640e7b9 |
| institution | Kabale University |
| issn | 2504-446X |
| language | English |
| publishDate | 2024-11-01 |
| publisher | MDPI AG |
| record_format | Article |
| series | Drones |
| spelling | doaj-art-8fec63294bbc45fc9cb09db80640e7b92024-11-26T18:00:43ZengMDPI AGDrones2504-446X2024-11-0181165310.3390/drones8110653LI-YOLO: An Object Detection Algorithm for UAV Aerial Images in Low-Illumination ScenesSongwen Liu0Hao He1Zhichao Zhang2Yatong Zhou3School of Electronic and Information Engineering, Hebei University of Technology, Tianjin 300401, ChinaSchool of Electronic and Information Engineering, Hebei University of Technology, Tianjin 300401, ChinaSchool of Electronic and Information Engineering, Hebei University of Technology, Tianjin 300401, ChinaSchool of Electronic and Information Engineering, Hebei University of Technology, Tianjin 300401, ChinaWith the development of unmanned aerial vehicle (UAV) technology, deep learning is becoming more and more widely used in object detection in UAV aerial images; however, detecting and identifying small objects in low-illumination scenes is still a major challenge. Aiming at the problem of low brightness, high noise, and obscure details of low-illumination images, an object detection algorithm, LI-YOLO (Low-Illumination You Only Look Once), for UAV aerial images in low-illumination scenes is proposed. Specifically, in the feature extraction section, this paper proposes a feature enhancement block (FEB) to realize global receptive field and context information learning through lightweight operations and embeds it into the C2f module at the end of the backbone network to alleviate the problems of high noise and detail blur caused by low illumination with very few parameter costs. In the feature fusion part, aiming to improve the detection performance for small objects in UAV aerial images, a shallow feature fusion network and a small object detection head are added. In addition, the adaptive spatial feature fusion structure (ASFF) is also introduced, which adaptively fuses information from different levels of feature maps by optimizing the feature fusion strategy so that the network can more accurately identify and locate objects of various scales. The experimental results show that the mAP50 of LI-YOLO reaches 76.6% on the DroneVehicle dataset and 90.8% on the LLVIP dataset. Compared with other current algorithms, LI-YOLO improves the mAP 50 by 3.1% on the DroneVehicle dataset and 6.9% on the LLVIP dataset. Experimental results show that the proposed algorithm can effectively improve object detection performance in low-illumination scenes.https://www.mdpi.com/2504-446X/8/11/653low illuminationsmall object detectionUAVYOLOv8 |
| spellingShingle | Songwen Liu Hao He Zhichao Zhang Yatong Zhou LI-YOLO: An Object Detection Algorithm for UAV Aerial Images in Low-Illumination Scenes Drones low illumination small object detection UAV YOLOv8 |
| title | LI-YOLO: An Object Detection Algorithm for UAV Aerial Images in Low-Illumination Scenes |
| title_full | LI-YOLO: An Object Detection Algorithm for UAV Aerial Images in Low-Illumination Scenes |
| title_fullStr | LI-YOLO: An Object Detection Algorithm for UAV Aerial Images in Low-Illumination Scenes |
| title_full_unstemmed | LI-YOLO: An Object Detection Algorithm for UAV Aerial Images in Low-Illumination Scenes |
| title_short | LI-YOLO: An Object Detection Algorithm for UAV Aerial Images in Low-Illumination Scenes |
| title_sort | li yolo an object detection algorithm for uav aerial images in low illumination scenes |
| topic | low illumination small object detection UAV YOLOv8 |
| url | https://www.mdpi.com/2504-446X/8/11/653 |
| work_keys_str_mv | AT songwenliu liyoloanobjectdetectionalgorithmforuavaerialimagesinlowilluminationscenes AT haohe liyoloanobjectdetectionalgorithmforuavaerialimagesinlowilluminationscenes AT zhichaozhang liyoloanobjectdetectionalgorithmforuavaerialimagesinlowilluminationscenes AT yatongzhou liyoloanobjectdetectionalgorithmforuavaerialimagesinlowilluminationscenes |