Improved Vehicle and Pedestrian Detection from UAV Perspective Based on Lightweight YOLOv8n
Facing the challenges of vehicle detection from a UAV perspective, such as small targets, occlusions, and dense arrangements, this paper proposes a lightweight vehicle and pedestrian detection algorithm based on YOLOv8n: FDT-YOLOv8. In the neck, the original PA-FPN structure is replaced with the Focusing Diffusion Pyramid Network (FDPN) to enrich the features of contextual information. Secondly, the Task Align Dynamic Detection Head (TADD-Head) is used to replace the original Decoupled-Head, which has been proven to enhance the detection head’s localization and classification performance. By employing shared convolutions, the number of parameters can be significantly reduced, making the model more lightweight. Experimental results show that the model’s parameter count is reduced by 13%, and [email protected] is increased by 4%, achieving the requirement of lightweight design.
Authors
- Haibo Li (ORCID: https://orcid.org/0000-0001-5969-9141)
- Ruixi Li (ORCID: https://orcid.org/0000-0002-0950-7532)
Institutions
- Shanghai University of Engineering Science (CN)
Publication Details
- Published
- 2024-11-01
- DOI
- https://doi.org/10.1109/iccbd-ai65562.2024.00052
- Primary Topic
- Advanced Neural Network Applications
- Type
- article
- Field-Weighted Citation Impact
- 0.00