A gated-attention multi-prior injection diffusion model for Huashan rock art image restoration
摘要
This paper addresses the problems of fading, loss, and texture degradation in the Zuojiang Huashan rock art caused by long term weathering and human disturbances, and constructs a Huashan rock art dataset for image restoration while proposing a prior guided restoration framework based on Stable Diffusion. The proposed method introduces a dual branch prior extraction module to separately model structural contours and texture material information during the diffusion denoising process, and employs a gate attention fusion module to achieve adaptive prior injection and weighted fusion, thereby improving structural continuity and material consistency in missing regions. At the data level, a total of 528 candidate images are collected, among which 177 valid samples are retained after unified screening and cleaning, and training and evaluation pairs are constructed using a unified 512