CrackSAM: Study on Few-Shot Segmentation of Mural Crack Images Using SAM
Due to the scarcity of mural disease data and the complexity of mural backgrounds, Traditional deep learning image segmentation requires a large amount of annotated data for training, which limits its segmentation performance. Although the SAM (Segment Anything Model) excels in natural image segmentation with its zero-shot segmentation capabilities and geometric prompts, it performs poorly in segmenting mural cracks and requires manual input prompts during the inference stage. This study proposes an novel mural crack segmentation method, CrackSAM. First, the high-performance image encoder from the SAM model is used to generate image embedding feature maps of the input images. Next, our designed feature extraction module further refines these features, enhancing the model's ability to capture crack details. Then, a decoder module upsamples and decodes the refined features to generate highresolution segmentation masks. This process ensures the accuracy of the segmentation results. In experiments, even with only a small amount of public crack datasets for training, our method demonstrates performance close to specialized crack segmentation networks trained on full datasets. Notably, on our self-constructed mural crack dataset, the method effectively distinguishes and segments cracks in complex backgrounds, proving its potential and effectiveness in practical applications. This achievement provides a new technical approach for the automated segmentation and assessment of mural damages.
Authors
- Haotong Zhang (ORCID: https://orcid.org/0000-0002-6617-5300)
- Xiaoping Jiang (ORCID: https://orcid.org/0000-0002-3588-1675)
- Shuai Zhang (ORCID: https://orcid.org/0000-0001-9153-5673)
- Shiqi Wang (ORCID: https://orcid.org/0000-0002-3583-959X)
- Chenghua Li (ORCID: https://orcid.org/0000-0002-6392-3342)
- Xu Han (ORCID: https://orcid.org/0000-0003-3707-660X)
Institutions
- Minzu University of China (CN)
- Dunhuang Research Academy (CN)
Publication Details
- Published
- 2024-08-16
- DOI
- https://doi.org/10.1109/icait62580.2024.10808070
- Citations
- 1
- Primary Topic
- Infrastructure Maintenance and Monitoring
- Type
- article
- Field-Weighted Citation Impact
- 0.26
Funders
- National Key Research and Development Program of China