YOLO-ROC轻量高效,专为实时道路损伤检测设计
YOLO-ROC: A High-Precision and Ultra-Lightweight Model for Real-Time Road Damage Detection
- 设计双向多尺度模块增强小目标特征提取能力
- 参数量降至0.89M,计算量降为2.6 GFLOPs,mAP50达67.6%
- 特别提升小目标检测效果,适合部署于移动端或边缘设备
道路损伤检测对保障交通安全和维护基础设施至关重要。尽管深度学习方法已被广泛采用,但仍面临两大挑战:现有网络多尺度特征提取能力不足,导致细小损伤(如裂缝、坑洼)漏检率高;主流模型参数量大、计算开销高,难以在实际场景中实现高效实时检测。为此,本文提出一种高精度轻量化模型 YOLO-ROC。通过设计双向多尺度空间金字塔池化快速模块(BMS-SPPF),结合双向空间-通道注意力机制,显著提升小目标检测能力;同时采用分层通道压缩策略,将参数量从3.01M降至0.89M,GFLOPs从8.1降至2.6。在RDD2022_China_Drone数据集上,YOLO-ROC的mAP50达到67.6%,较YOLOv8n提升2.11%;小目标类别D40的mAP50提升16.8%,模型大小仅2.0 MB。此外,在RDD2022_China_Motorbike数据集上也表现出优异泛化性能。
原文摘要 · Abstract (English)
Road damage detection is a critical task for ensuring traffic safety and maintaining infrastructure integrity. While deep learning-based detection methods are now widely adopted, they still face two core challenges: first, the inadequate multi-scale feature extraction capabilities of existing networks for diverse targets like cracks and potholes, leading to high miss rates for small-scale damage; and second, the substantial parameter counts and computational demands of mainstream models, which hinder their deployment for efficient, real-time detection in practical applications. To address these issues, this paper proposes a high-precision and lightweight model, YOLO - Road Orthogonal Compact (YOLO-ROC). We designed a Bidirectional Multi-scale Spatial Pyramid Pooling Fast (BMS-SPPF) module to enhance multi-scale feature extraction and implemented a hierarchical channel compression strategy to reduce computational complexity. The BMS-SPPF module leverages a bidirectional spatial-channel attention mechanism to improve the detection of small targets. Concurrently, the channel compression strategy reduces the parameter count from 3.01M to 0.89M and GFLOPs from 8.1 to 2.6. Experiments on the RDD2022_China_Drone dataset demonstrate that YOLO-ROC achieves a mAP50 of 67.6%, surpassing the baseline YOLOv8n by 2.11%. Notably, the mAP50 for the small-target D40 category improved by 16.8%, and the final model size is only 2.0 MB. Furthermore, the model exhibits excellent generalization performance on the RDD2022_China_Motorbike dataset.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。