用图像标签实现裂缝精准定位,无需逐像素标注。
WP-CrackNet: A Collaborative Adversarial Learning Framework for End-to-End Weakly-Supervised Road Crack Detection
- 三模块协同对抗学习,自动生成高质量伪标签。
- 在三个新构建数据集上达到监督方法水平,显著优于现有弱监督模型。
- 适合智能交通巡检、城市基建维护等场景应用。
道路裂缝检测对智慧城市建设至关重要。为降低对昂贵像素级标注的依赖,我们提出WP-CrackNet,一种仅需图像级标签即可实现像素级裂缝检测的端到端弱监督方法。该方法整合分类器(生成类激活图CAM)、重构器(衡量特征可推断性)与检测器三组件,通过交替对抗训练使裂纹CAM覆盖完整裂纹区域,检测器则基于后处理后的裂纹CAM生成的伪标签进行学习。三者互馈提升学习稳定性和检测精度。进一步设计路径感知注意力模块(PAAM),融合分类器高层语义与重构器低层结构线索,建模空间与通道依赖。提出中心增强的CAM一致性模块(CECCM),结合中心高斯加权与一致性约束,优化裂纹CAM,提升伪标签质量。构建三个图像级标注数据集,大量实验表明,WP-CrackNet性能媲美监督方法,显著超越现有弱监督方法,推动道路检测规模化发展。源代码与数据集见https://mias.group/WP-CrackNet/。
原文摘要 · Abstract (English)
Road crack detection is essential for intelligent infrastructure maintenance in smart cities. To reduce reliance on costly pixel-level annotations, we propose WP-CrackNet, an end-to-end weakly-supervised method that trains with only image-level labels for pixel-wise crack detection. WP-CrackNet integrates three components: a classifier generating class activation maps (CAMs), a reconstructor measuring feature inferability, and a detector producing pixel-wise road crack detection results. During training, the classifier and reconstructor alternate in adversarial learning to encourage crack CAMs to cover complete crack regions, while the detector learns from pseudo labels derived from post-processed crack CAMs. This mutual feedback among the three components improves learning stability and detection accuracy. To further boost detection performance, we design a path-aware attention module (PAAM) that fuses high-level semantics from the classifier with low-level structural cues from the reconstructor by modeling spatial and channel-wise dependencies. Additionally, a center-enhanced CAM consistency module (CECCM) is proposed to refine crack CAMs using center Gaussian weighting and consistency constraints, enabling better pseudo-label generation. We create three image-level datasets and extensive experiments show that WP-CrackNet achieves comparable results to supervised methods and outperforms existing weakly-supervised methods, significantly advancing scalable road inspection. The source code package and datasets are available at https://mias.group/WP-CrackNet/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。