pytorch implementation of the paper ``Large Scale Image Completion via Co-Modulated Generative Adversarial Networks"
Implementation of the paper ``Large Scale Image Completion via Co-Modulated Generative Adversarial Networks"
official tensorflow version: https://github.com/zsyzzsoft/co-mod-gan
Input image Mask Result
conda install pytorch torchvision cudatoolkit=11 -c pytorch
conda install matplotlib jinja2 ninja dill
pip install git+https://github.com/zengxianyu/pytorch-fid
Download the code:
git clone https://github.com/zengxianyu/co-mod-gan-pytorch
git checkout train
git submodule init
git submodule update
e.g. ffhq512
./download/ffhq512.sh
converted model:
./test.sh
./download/data.sh
./train.sh
Coming soon
[1] official tensorflow version: https://github.com/zsyzzsoft/co-mod-gan
[2] stylegan2-pytorch https://github.com/rosinality/stylegan2-pytorch
[3] pix2pixHD https://github.com/NVIDIA/pix2pixHD
[4] SPADE https://github.com/NVlabs/SPADE