Retrieval Augmented Image Harmonization

This is the official repository for our paper: Retrieval Augmented Image Harmonization (Accpeted by PR 2025)

https://arxiv.org/abs/2412.13916

Pre-train Models

Download the pretrained model from the following url：

• BaiduNetDisk

Quick Training

bash train.sh

Testing

bash test.sh

Acknowledgement

This project is built based on the excellent [HDNet] (https://github.com/chenhaoxing/HDNet) and Denoising ViT (https://github.com/Jiawei-Yang/Denoising-ViT)

Citation

@article{wang2024retrieval,
  title={Retrieval Augmented Image Harmonization},
  author={Wang, Haolin and Liu, Ming and Yan, Zifei and Zhou, Chao and Xiao, Longan and Zuo, Wangmeng},
  journal={arXiv preprint arXiv:2412.13916},
  year={2024}
}