PPD-YOLO: A lightweight model for small target detection in extreme light and dense scenes for UAVs
摘要
In UAV applications, detecting small targets is challenging due to the presence of dense targets and extreme lighting conditions. This paper proposes PPD-YOLO, an efficient model for robust detection in these challenging scenes. First, two specialized detection heads are designed to capture finer features and enhance performance for detecting small targets. Second, a PP-FPN network structure is proposed to replace the traditional PAFPN, aiming to retain and convey information better about small targets. Finally, the C2f-DW module is designed to replace part of the C2f module further enabling a lightweight model. On the VisDrone2019 dataset, PPD-YOLO achieves an mAP50 of 46.6% with only 5.7 million parameters, surpassing the baseline by a significant margin of 7.5 percentage points while reducing the parameter count by 5.4 million. On the TinyPerson dataset, it also attains an mAP50 of 22.7%. These results demonstrate that PPD-YOLO outperforms existing YOLO series methods for small object detection, excelling in both detection accuracy and model compactness.