peterzhousz / ssharmonization

[ICCV'2021] "SSH: A Self-Supervised Framework for Image Harmonization", Yifan Jiang, He Zhang, Jianming Zhang, Yilin Wang, Zhe Lin, Kalyan Sunkavalli, Simon Chen, Sohrab Amirghodsi, Sarah Kong, Zhangyang Wang

SSH: A Self-Supervised Framework for Image Harmonization (ICCV 2021)

We provide the inference code and collected Dataset for Self-supervised Image Harmonization

Guideline

Download the RealHM dataset and put it on ../RealH', then download the pretrained weight and put it on ./checkpoint. Then Run the demo.ipynb.

updates

Bugs are fixed in the newest version (Jan. 2022)

Representative Examples

Main Pipeline

Dataset Pipeline

Citation

@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}
}