Chen Weijia, Liu Jiaming, Liu Tong, Zhuang Yaoming
Faculty of Business Administration, Northeastern University, Shenyang, 110819, China.
Faculty of Robot Science and Engineering, Northeastern University, Shenyang, 110819, China.
Sci Rep. 2025 Aug 16;15(1):29988. doi: 10.1038/s41598-025-15975-w.
In the domain of object detection, small object detection remains a pressing challenge, as existing approaches often suffer from limited accuracy, high model complexity, and difficulty meeting lightweight deployment requirements. In this paper, we propose PCPE-YOLO, a novel object detection algorithm, specifically designed to address these difficulties. First, we put forward a dynamically reconfigurable C2f_PIG module. This module uses a parameter-aware mechanism to adapt its bottleneck structures to different network depths and widths, reducing parameters while maintaining performance. Next, we introduce a Context Anchor Attention mechanism that boosts the model's focus on the contexts of small objects, thereby improving detection accuracy. In addition, we add a small object detection layer to enhance the model's localization capability for small objects. Finally, we integrate an Efficient Up-Convolution Block to sharpen decoder feature maps, enhancing small object recall with minimal computational overhead. Experiments on VisDrone2019, KITTI, and NWPU VHR-10 datasets show that PCPE-YOLO significantly outperforms both the baseline and other state-of-the-art methods in precision, recall, mean average precision, and parameters, achieving the best precision among all compared approaches. On VisDrone2019 in particular, it achieves improvements of 3.8% in precision, 5.6% in recall, 6.2% in mAP50, and 5% in F1 score, effectively combining lightweight design with high small object detection performance and providing a more efficient and reliable solution for small object detection in real-world applications.
在目标检测领域,小目标检测仍然是一个紧迫的挑战,因为现有方法往往存在精度有限、模型复杂度高以及难以满足轻量级部署要求等问题。在本文中,我们提出了PCPE-YOLO,一种专门设计用于解决这些难题的新型目标检测算法。首先,我们提出了一种动态可重构的C2f_PIG模块。该模块使用参数感知机制,使其瓶颈结构能够适应不同的网络深度和宽度,在保持性能的同时减少参数。接下来,我们引入了上下文锚点注意力机制,增强模型对小目标上下文的关注,从而提高检测精度。此外,我们添加了一个小目标检测层,以增强模型对小目标的定位能力。最后,我们集成了一个高效上卷积块,以锐化解码器特征图,以最小的计算开销提高小目标召回率。在VisDrone2019、KITTI和NWPU VHR-10数据集上的实验表明,PCPE-YOLO在精度、召回率、平均精度均值和参数方面均显著优于基线方法和其他现有最先进方法,在所有比较方法中实现了最佳精度。特别是在VisDrone2019上,它在精度上提高了3.8%,召回率上提高了5.6%,mAP50上提高了6.2%,F1分数上提高了5%,有效地将轻量级设计与高小目标检测性能相结合,为实际应用中的小目标检测提供了更高效可靠的解决方案。