arXiv:2501.02822cs.CVcs.AI2025-01被引 6

提出新数据集与4D注意力模型,提升多尺度道路损伤检测精度

RDD4D: 4D Attention-Guided Road Damage Detection And Classification

  • 设计Attention4D模块融合位置编码与全局上下文感知
  • 在自建数据集上大裂缝检测AP达0.458,整体AP为0.445
  • 适合道路智能巡检与交通运维系统研发人员参考

道路损伤检测与评估是基础设施维护的关键环节。现有方法在单张图像中检测多种类型、不同尺度的损伤时表现不佳,主要因缺乏涵盖多样损伤类型且尺度各异的数据集。为此,我们首先构建了全新的多样化道路损伤数据集(DRDD),填补了当前数据集的空白。随后提出RDD4D模型,采用Attention4D模块,通过结合位置编码与“Talking Head”结构的注意力机制,实现多尺度特征优化。在自建数据集上的实验表明,该模型对大尺寸裂缝检测的平均精度(AP)达到0.458,整体AP为0.445。此外,在CrackTinyNet数据集上性能提升约0.21。代码、权重、数据集及结果均已开源。

原文摘要 · Abstract (English)

Road damage detection and assessment are crucial components of infrastructure maintenance. However, current methods often struggle with detecting multiple types of road damage in a single image, particularly at varying scales. This is due to the lack of road datasets with various damage types having varying scales. To overcome this deficiency, first, we present a novel dataset called Diverse Road Damage Dataset (DRDD) for road damage detection that captures the diverse road damage types in individual images, addressing a crucial gap in existing datasets. Then, we provide our model, RDD4D, that exploits Attention4D blocks, enabling better feature refinement across multiple scales. The Attention4D module processes feature maps through an attention mechanism combining positional encoding and "Talking Head" components to capture local and global contextual information. In our comprehensive experimental analysis comparing various state-of-the-art models on our proposed, our enhanced model demonstrated superior performance in detecting large-sized road cracks with an Average Precision (AP) of 0.458 and maintained competitive performance with an overall AP of 0.445. Moreover, we also provide results on the CrackTinyNet dataset; our model achieved around a 0.21 increase in performance. The code, model weights, dataset, and our results are available on \href{https://github.com/msaqib17/Road_Damage_Detection}{https://github.com/msaqib17/Road\_Damage\_Detection}.

道路检测多尺度注意力机制数据集

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。