YOLO-WSR: A wavelet feature and structure guided diffusion model for detection and automated restoration of cultural heritage artifact images
- 1Academy of Fine Arts and Crafts, Zhejiang Guangsha Vocational and Technical University of Construction, Dongyang, China
- 2Zhejiang Guangsha Vocational and Technical University of Construction, Dongyang, China
- 3Advanced Institute of Information Technology Peking University, Dongyang, China
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.
Related
- Towards Automated Acoustic Monitoring of Threatened Birds in Tropical Forest Ecosystem Restoration
- Migrations facilitated the cultural diffusion of megaliths in Neolithic Europe
- Feasibility study of fluid-structure-growth modelling in the context of endogenous tissue restoration
- Restoration pathways reshape forest structure and multifunctionality in subtropical plantations and secondary forests
- Evidence for a cohesion-related function of alpaca hums: Linking automated call detection with GPS data
BibTeX
@article{chen2026yolowsr,
title = {{YOLO-WSR: A wavelet feature and structure guided diffusion model for detection and automated restoration of cultural heritage artifact images}},
author = {Yi Chen and Xing Wang and Wang Shaohua},
journal = {PLOS One},
year = {2026},
month = sep,
doi = {10.1371/journal.pone.0351571},
url = {https://doi.org/10.1371/journal.pone.0351571},
note = {CC BY 4.0},
}
RIS
TY - JOUR
TI - YOLO-WSR: A wavelet feature and structure guided diffusion model for detection and automated restoration of cultural heritage artifact images
AU - Yi Chen
AU - Xing Wang
AU - Wang Shaohua
PY - 2026
DA - 2026/09/11
JO - PLOS One
DO - 10.1371/journal.pone.0351571
UR - https://doi.org/10.1371/journal.pone.0351571
AB - 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.
SN - 1932-6203
ER -