Skip to content

[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

License

Notifications You must be signed in to change notification settings

VITA-Group/SSHarmonization

Repository files navigation

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

Visual_Examples

Main Pipeline

Pipeline

Dataset Pipeline

Editor

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

About

[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

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published