VITA-Group / 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'. Ddownload the pretrained weight and put it on ./.
Create directory output. Then Run the demo.ipynb.

pip install -r requirements.txt

updates

add environment file (Feb. 2022) Bugs are fixed in the newest version (Jan. 2022)

Representative Examples

Main Pipeline

Dataset Pipeline

Citation

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