优化YOLO系列模型,提升道路破损检测速度与准确率。
Optimizing YOLO Architectures for Optimal Road Damage Detection and Classification: A Comparative Study from YOLOv7 to YOLOv10
- 用改进的YOLOv7和Tiny YOLOv7组合,结合注意力机制提升检测力。
- 在0.0547秒/图的推理速度下,达到0.7027的F1分数。
- 适合交通基础设施智能巡检、边缘设备部署场景使用。
维护道路基础设施对保障交通安全、高效和可持续交通系统至关重要。然而,人工采集道路损伤数据耗时费力且存在安全隐患。近年来,人工智能特别是深度学习为利用道路图像自动化检测提供了可行方案。本文提出一种完整的道路损伤检测深度学习工作流程,重点优化推理速度同时保持检测精度。为适应硬件限制,采用大图切块并使用轻量级模型;另引入外部坑洼数据集以增强该类损伤的检测能力。所提方法采用多个模型架构,包括含坐标注意力层的自定义YOLOv7模型和默认的Tiny YOLOv7模型,经训练后融合以最大化性能。模型进一步重参数化以优化推理效率。实验表明,含三个坐标注意力层的自定义YOLOv7与默认Tiny YOLOv7模型的集成,在0.0547秒/图的推理速度下实现0.7027的F1分数。完整流程(含数据预处理、模型训练与推理脚本)已公开于项目GitHub仓库,支持可复现性并促进后续研究。
原文摘要 · Abstract (English)
Maintaining roadway infrastructure is essential for ensuring a safe, efficient, and sustainable transportation system. However, manual data collection for detecting road damage is time-consuming, labor-intensive, and poses safety risks. Recent advancements in artificial intelligence, particularly deep learning, offer a promising solution for automating this process using road images. This paper presents a comprehensive workflow for road damage detection using deep learning models, focusing on optimizations for inference speed while preserving detection accuracy. Specifically, to accommodate hardware limitations, large images are cropped, and lightweight models are utilized. Additionally, an external pothole dataset is incorporated to enhance the detection of this underrepresented damage class. The proposed approach employs multiple model architectures, including a custom YOLOv7 model with Coordinate Attention layers and a Tiny YOLOv7 model, which are trained and combined to maximize detection performance. The models are further reparameterized to optimize inference efficiency. Experimental results demonstrate that the ensemble of the custom YOLOv7 model with three Coordinate Attention layers and the default Tiny YOLOv7 model achieves an F1 score of 0.7027 with an inference speed of 0.0547 seconds per image. The complete pipeline, including data preprocessing, model training, and inference scripts, is publicly available on the project's GitHub repository, enabling reproducibility and facilitating further research.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。