📋 Mural Restoration

CMAMRNet: A Contextual Mask-Aware Network Enhancing Mural Restoration Through Comprehensive Mask Guidance

Yingtie Lei, Fanghai Yi, Yihang Dong, Weihuang Liu, Xiaofeng Zhang, Zimeng Li, Chi-Man Pun and Xuhang Chen 📮 (📮 Corresponding authors)

University of Macau, Guangdong University of Technology, University of Chinese Academy of Sciences, Shanghai Jiao Tong University, Shenzhen Polytechnic University, Huizhou University

2025 British Machine Vision Conference (BMVC 2025)

🔮 Dataset

The dataset is available at Kaggle.

⚙️ Usage

Training

You may download the dataset first, and then specify TRAIN_DIR, VAL_DIR and SAVE_DIR in the section TRAINING in config.yml.

For single GPU training:

python train.py

For multiple GPUs training:

accelerate config
accelerate launch train.py

If you have difficulties with the usage of accelerate, please refer to Accelerate.

Inference

python test.py

💗 Acknowledgement

This work was supported in part by the Science and Technology Development Fund, Macau SAR, under Grant 0141/2023/RIA2 and 0193/2023/RIA3, and the University of Macau under Grant MYRG-GRG2024-00065-FST-UMDF, in part by the Shenzhen Polytechnic University Research Fund (Grant No. 6025310023K).

🛎 Citation

If you find our work helpful for your research, please cite:

@inproceedings{Lei_2025_BMVC,
  author    = {Yingtie Lei and Fanghai Yi and Yihang Dong and Weihuang Liu and Xiaofeng Zhang and Zimeng Li and Chi-Man Pun and Xuhang Chen},
  title     = {CMAMRNet: A Contextual Mask-Aware Network Enhancing Mural Restoration Through Comprehensive Mask Guidance},
  booktitle = {36th British Machine Vision Conference 2025, {BMVC} 2025, Sheffield, UK, November 24-27, 2025},
  publisher = {BMVA},
  year      = {2025},
}