We provide the inference code and collected Dataset for Self-supervised Image Harmonization
Download the RealHM dataset and put it on ../RealH
', then download the pretrained weight and put it on ./checkpoint
. Then Run the demo.ipynb
.
Bugs are fixed in the newest version (Jan. 2022)
@article{jiang2021ssh,
title={SSH: A Self-Supervised Framework for Image Harmonization},
author={Jiang, Yifan and Zhang, He and Zhang, Jianming and Wang, Yilin and Lin, Zhe and Sunkavalli, Kalyan and Chen, Simon and Amirghodsi, Sohrab and Kong, Sarah and Wang, Zhangyang},
journal={arXiv preprint arXiv:2108.06805},
year={2021}
}