arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.04786cs.CV
收藏

SOD-YOLOv8——增强YOLOv8以用于交通场景中的小目标检测

SOD-YOLOv8 -- Enhancing YOLOv8 for Small Object Detection in Traffic Scenes

Columbia University(哥伦比亚大学)

Boshra Khalili, Andrew W. Smyth

AI总结:

本文提出SOD-YOLOv8,通过GFPN式多路径特征融合、第四检测层、C2f-EMA注意力和PIoU损失增强YOLOv8,在不显著增加成本的情况下提升交通场景小目标检测性能。

摘要:

目标检测作为计算机视觉的一部分,对于交通管理、应急响应、自动驾驶汽车和智慧城市可能至关重要。尽管目标检测取得了重大进展,但在远距离相机拍摄的图像中检测小目标仍具挑战性,原因在于其尺寸、与相机的距离、多样的形状以及杂乱的背景。为应对这些挑战,我们提出了Small Object Detection YOLOv8(SOD-YOLOv8),这是一种专门为涉及大量小目标的场景设计的新型模型。受Efficient Generalized Feature Pyramid Networks(GFPN)启发,我们增强了YOLOv8内部的多路径融合,以整合不同层级的特征,保留来自较浅层的细节并提高小目标检测精度。此外,模型增加了第四个检测层,以有效利用高分辨率空间信息。C2f-EMA模块中的Efficient Multi-Scale Attention Module(EMA)通过重新分配权重并优先处理相关特征来增强特征提取。我们引入Powerful-IoU(PIoU)替代CIoU,其聚焦于中等质量的锚框,并基于预测边界框与真实边界框角点之间的差异增加惩罚项。该方法简化了计算、加快了收敛并提升了检测精度。SOD-YOLOv8显著改善了小目标检测,在多项指标上超过了广泛使用的模型,同时与YOLOv8s相比没有大幅增加计算成本或延迟。具体而言,它将召回率从40.1%提高到43.9%,精确率从51.2%提高到53.9%,mAP₀.₅从40.6%提高到45.1%,mAP₀.₅:₀.₉₅从24%提高到26.6%。在动态的真实交通场景中,SOD-YOLOv8在多种条件下展现出显著提升,证明了其即使在具有挑战性的环境中检测小目标的可靠性和有效性。

英文摘要:

Object detection as part of computer vision can be crucial for traffic management, emergency response, autonomous vehicles, and smart cities. Despite significant advances in object detection, detecting small objects in images captured by distant cameras remains challenging due to their size, distance from the camera, varied shapes, and cluttered backgrounds. To address these challenges, we propose Small Object Detection YOLOv8 (SOD-YOLOv8), a novel model specifically designed for scenarios involving numerous small objects. Inspired by Efficient Generalized Feature Pyramid Networks (GFPN), we enhance multi-path fusion within YOLOv8 to integrate features across different levels, preserving details from shallower layers and improving small object detection accuracy. Also, A fourth detection layer is added to leverage high-resolution spatial information effectively. The Efficient Multi-Scale Attention Module (EMA) in the C2f-EMA module enhances feature extraction by redistributing weights and prioritizing relevant features. We introduce Powerful-IoU (PIoU) as a replacement for CIoU, focusing on moderate-quality anchor boxes and adding a penalty based on differences between predicted and ground truth bounding box corners. This approach simplifies calculations, speeds up convergence, and enhances detection accuracy. SOD-YOLOv8 significantly improves small object detection, surpassing widely used models in various metrics, without substantially increasing computational cost or latency compared to YOLOv8s. Specifically, it increases recall from 40.1\% to 43.9\%, precision from 51.2\% to 53.9\%, $\text{mAP}_{0.5}$ from 40.6\% to 45.1\%, and $\text{mAP}_{0.5:0.95}$ from 24\% to 26.6\%. In dynamic real-world traffic scenes, SOD-YOLOv8 demonstrated notable improvements in diverse conditions, proving its reliability and effectiveness in detecting small objects even in challenging environments.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑