├── .gitignore ├── .gitmodules ├── LICENSE ├── README.md ├── data ├── download_eval_data.sh ├── download_lsun.sh ├── download_sketch_data.sh └── prepare_lsun.py ├── eval ├── __init__.py ├── evaluation.py └── ppl.py ├── ganspace.py ├── generate.py ├── images ├── cat1.jpg ├── cat2.jpg ├── editing.jpg ├── failure_case.jpg ├── interp2.gif ├── model_interp │ ├── 0.png │ ├── 1.png │ ├── 2.png │ ├── latent_1.gif │ ├── latent_2.gif │ ├── weight_1.gif │ └── weight_2.gif ├── teaser.jpg └── teaser_video.gif ├── options.py ├── pretrained └── download_pretrained_models.sh ├── quickstart_colab.ipynb ├── requirements.txt ├── run_metrics.py ├── scripts ├── train_authorsketch_ffhq0.sh ├── train_photosketch_horse_riders.sh ├── train_quickdraw_single_horse0.sh └── train_teaser_cat.sh ├── train.py ├── training ├── __init__.py ├── dataset.py ├── gan_model.py ├── gan_trainer.py └── networks │ ├── __init__.py │ ├── diffaug.py │ ├── loss.py │ ├── misc.py │ ├── op │ ├── __init__.py │ ├── fused_act.py │ ├── fused_bias_act.cpp │ ├── fused_bias_act_kernel.cu │ ├── upfirdn2d.cpp │ ├── upfirdn2d.py │ └── upfirdn2d_kernel.cu │ ├── pix2pix.py │ ├── stylegan2.py │ └── transform.py ├── util ├── __init__.py ├── html.py ├── util.py └── visualizer.py └── weights ├── download_weights.sh ├── ganspace_cat.npz ├── ganspace_church.npz └── ganspace_horse.npz /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/.gitmodules -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/README.md -------------------------------------------------------------------------------- /data/download_eval_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/data/download_eval_data.sh -------------------------------------------------------------------------------- /data/download_lsun.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/data/download_lsun.sh -------------------------------------------------------------------------------- /data/download_sketch_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/data/download_sketch_data.sh -------------------------------------------------------------------------------- /data/prepare_lsun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/data/prepare_lsun.py -------------------------------------------------------------------------------- /eval/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/eval/__init__.py -------------------------------------------------------------------------------- /eval/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/eval/evaluation.py -------------------------------------------------------------------------------- /eval/ppl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/eval/ppl.py -------------------------------------------------------------------------------- /ganspace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/ganspace.py -------------------------------------------------------------------------------- /generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/generate.py -------------------------------------------------------------------------------- /images/cat1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/images/cat1.jpg -------------------------------------------------------------------------------- /images/cat2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/images/cat2.jpg -------------------------------------------------------------------------------- /images/editing.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/images/editing.jpg -------------------------------------------------------------------------------- /images/failure_case.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/images/failure_case.jpg -------------------------------------------------------------------------------- /images/interp2.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/images/interp2.gif -------------------------------------------------------------------------------- /images/model_interp/0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/images/model_interp/0.png -------------------------------------------------------------------------------- /images/model_interp/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/images/model_interp/1.png -------------------------------------------------------------------------------- /images/model_interp/2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/images/model_interp/2.png -------------------------------------------------------------------------------- /images/model_interp/latent_1.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/images/model_interp/latent_1.gif -------------------------------------------------------------------------------- /images/model_interp/latent_2.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/images/model_interp/latent_2.gif -------------------------------------------------------------------------------- /images/model_interp/weight_1.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/images/model_interp/weight_1.gif -------------------------------------------------------------------------------- /images/model_interp/weight_2.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/images/model_interp/weight_2.gif -------------------------------------------------------------------------------- /images/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/images/teaser.jpg -------------------------------------------------------------------------------- /images/teaser_video.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/images/teaser_video.gif -------------------------------------------------------------------------------- /options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/options.py -------------------------------------------------------------------------------- /pretrained/download_pretrained_models.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/pretrained/download_pretrained_models.sh -------------------------------------------------------------------------------- /quickstart_colab.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/quickstart_colab.ipynb -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/run_metrics.py -------------------------------------------------------------------------------- /scripts/train_authorsketch_ffhq0.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/scripts/train_authorsketch_ffhq0.sh -------------------------------------------------------------------------------- /scripts/train_photosketch_horse_riders.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/scripts/train_photosketch_horse_riders.sh -------------------------------------------------------------------------------- /scripts/train_quickdraw_single_horse0.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/scripts/train_quickdraw_single_horse0.sh -------------------------------------------------------------------------------- /scripts/train_teaser_cat.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/scripts/train_teaser_cat.sh -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/train.py -------------------------------------------------------------------------------- /training/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /training/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/training/dataset.py -------------------------------------------------------------------------------- /training/gan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/training/gan_model.py -------------------------------------------------------------------------------- /training/gan_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/training/gan_trainer.py -------------------------------------------------------------------------------- /training/networks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/training/networks/__init__.py -------------------------------------------------------------------------------- /training/networks/diffaug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/training/networks/diffaug.py -------------------------------------------------------------------------------- /training/networks/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/training/networks/loss.py -------------------------------------------------------------------------------- /training/networks/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/training/networks/misc.py -------------------------------------------------------------------------------- /training/networks/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/training/networks/op/__init__.py -------------------------------------------------------------------------------- /training/networks/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/training/networks/op/fused_act.py -------------------------------------------------------------------------------- /training/networks/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/training/networks/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /training/networks/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/training/networks/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /training/networks/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/training/networks/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /training/networks/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/training/networks/op/upfirdn2d.py -------------------------------------------------------------------------------- /training/networks/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/training/networks/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /training/networks/pix2pix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/training/networks/pix2pix.py -------------------------------------------------------------------------------- /training/networks/stylegan2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/training/networks/stylegan2.py -------------------------------------------------------------------------------- /training/networks/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/training/networks/transform.py -------------------------------------------------------------------------------- /util/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /util/html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/util/html.py -------------------------------------------------------------------------------- /util/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/util/util.py -------------------------------------------------------------------------------- /util/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/util/visualizer.py -------------------------------------------------------------------------------- /weights/download_weights.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/weights/download_weights.sh -------------------------------------------------------------------------------- /weights/ganspace_cat.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/weights/ganspace_cat.npz -------------------------------------------------------------------------------- /weights/ganspace_church.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/weights/ganspace_church.npz -------------------------------------------------------------------------------- /weights/ganspace_horse.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeterWang512/GANSketching/HEAD/weights/ganspace_horse.npz --------------------------------------------------------------------------------