Weakly supervised pavement crack semantic segmentation based on multi-scale object localization and incremental annotation refinement

Zaid Al-Huda,Bo Peng,Riyadh Nazar Ali Algburi,Saghir Alfasly,Tianrui Li
DOI: https://doi.org/10.1007/s10489-022-04212-w
IF: 5.3
2022-10-28
Applied Intelligence
Abstract:Automatic and accurate pavement crack detection is essential for cost-effective road maintenance. Deep convolutional neural networks (DCNNs) are widely used in recent methods for pavement crack segmentation. Although DCNNs can segment pavement cracks with great accuracy, the requirement for huge pixel-level labels is demanding. In this article, we propose a novel weakly supervised framework for pavement crack segmentation based on multi-scale object localization and incremental annotation refinement. A trained pavement crack classification network is used to produce initial annotations using multi-scale class activation mapping strategy. Then, a new segmentation network (U 2 -Net) with triplet attention (TA) module and multiple loss functions is trained using initial annotations. The TA module is developed to emphasize important features and ignore unimportant features, whereas multiple loss functions are employed to assist crack segmentation for a clean and full mask. Moreover, incremental annotation refinement (IAR) is proposed for iteratively optimizing the segmentation network and refining segmentation masks. Comparative experiments on DeepCrack and Crack500 datasets demonstrate that the proposed framework bridges the performance gap between weakly and fully supervised pavement crack segmentation methods, outperforms existing weakly supervised pavement crack segmentation methods, and achieves state-of-the-art performance while reducing human labeling efforts.
computer science, artificial intelligence
What problem does this paper attempt to address?