YOLO-WSR: A wavelet feature and structure guided diffusion model for detection and automated restoration of cultural heritage artifact images

Yi Chen, Xing Wang and Wang Shaohua

PLOS ONE, 2026, vol. 21, issue 9, 1-29

Abstract: In response to the complexity and diversity involved in damage detection and automated restoration of cultural heritage artifact images associated with intangible cultural heritage preservation, this study proposes a YOLO-based restoration framework that integrates wavelet features with structure-guided diffusion, termed YOLO-WSR. The proposed method enhances detection accuracy through wavelet feature upsampling and employs a structure-guided diffusion generation mechanism to achieve effective restoration of texture and structural details in damaged artifact images. In addition, a CLIP-IQA-based feedback optimization module is introduced to perform no-reference quality assessment on the restored outputs and iteratively refine the restoration results, thereby forming a closed-loop collaborative framework that integrates detection, restoration, and evaluation. Experiments conducted on multiple publicly available datasets demonstrate the effectiveness of YOLO-WSR. The results show that the proposed method consistently improves performance over existing approaches, achieving approximately 5–6% improvement in mAP@50 for damage detection, while obtaining enhanced PSNR, SSIM, LPIPS, and SRCC performance for restoration quality evaluation. Further ablation studies demonstrate the contribution of each component to the overall framework. The proposed method provides an effective and quantifiable technical solution for the digital preservation of cultural heritage artifact images, offering practical support for accurate damage identification, automated restoration, and intelligent management of visual cultural heritage resources.

Date: 2026
References: Add references at CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0351571 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 51571&type=printable (application/pdf)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:plo:pone00:0351571

DOI: 10.1371/journal.pone.0351571

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().