We provide the inference code and collected Dataset for Self-supervised Image Harmonization
Download the RealHM dataset and put it on ../RealH
'.
Ddownload the pretrained weight and put it on ./
.
Create directory output
.
Then Run the demo.ipynb
.
pip install -r requirements.txt
add environment file (Feb. 2022) Bugs are fixed in the newest version (Jan. 2022)
@inproceedings{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},
booktitle={Proceedings of the IEEE/CVF International Conference on Computer Vision},
pages={4832--4841},
year={2021}
}