├── .gitignore ├── LICENSE ├── README.md ├── assets ├── acgan.gif ├── bicyclegan.png ├── bicyclegan_architecture.jpg ├── cgan.gif ├── cluster_gan.gif ├── cogan.gif ├── context_encoder.png ├── cyclegan.png ├── dcgan.gif ├── discogan.png ├── enhanced_superresgan.png ├── esrgan.png ├── gan.gif ├── infogan.gif ├── infogan.png ├── logo.png ├── munit.png ├── pix2pix.png ├── pixelda.png ├── stargan.png ├── superresgan.png ├── wgan_div.gif ├── wgan_div.png └── wgan_gp.gif ├── data ├── download_cyclegan_dataset.sh └── download_pix2pix_dataset.sh ├── implementations ├── aae │ └── aae.py ├── acgan │ └── acgan.py ├── began │ └── began.py ├── bgan │ └── bgan.py ├── bicyclegan │ ├── bicyclegan.py │ ├── datasets.py │ └── models.py ├── ccgan │ ├── ccgan.py │ ├── datasets.py │ └── models.py ├── cgan │ └── cgan.py ├── cluster_gan │ └── clustergan.py ├── cogan │ ├── cogan.py │ └── mnistm.py ├── context_encoder │ ├── context_encoder.py │ ├── datasets.py │ └── models.py ├── cyclegan │ ├── cyclegan.py │ ├── datasets.py │ ├── models.py │ └── utils.py ├── dcgan │ └── dcgan.py ├── discogan │ ├── datasets.py │ ├── discogan.py │ └── models.py ├── dragan │ └── dragan.py ├── dualgan │ ├── datasets.py │ ├── dualgan.py │ └── models.py ├── ebgan │ └── ebgan.py ├── esrgan │ ├── datasets.py │ ├── esrgan.py │ ├── models.py │ └── test_on_image.py ├── gan │ └── gan.py ├── infogan │ └── infogan.py ├── lsgan │ └── lsgan.py ├── munit │ ├── datasets.py │ ├── models.py │ └── munit.py ├── pix2pix │ ├── datasets.py │ ├── models.py │ └── pix2pix.py ├── pixelda │ ├── mnistm.py │ └── pixelda.py ├── relativistic_gan │ └── relativistic_gan.py ├── sgan │ └── sgan.py ├── softmax_gan │ └── softmax_gan.py ├── srgan │ ├── datasets.py │ ├── models.py │ └── srgan.py ├── stargan │ ├── datasets.py │ ├── models.py │ └── stargan.py ├── unit │ ├── datasets.py │ ├── models.py │ └── unit.py ├── wgan │ └── wgan.py ├── wgan_div │ └── wgan_div.py └── wgan_gp │ └── wgan_gp.py └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/README.md -------------------------------------------------------------------------------- /assets/acgan.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/acgan.gif -------------------------------------------------------------------------------- /assets/bicyclegan.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/bicyclegan.png -------------------------------------------------------------------------------- /assets/bicyclegan_architecture.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/bicyclegan_architecture.jpg -------------------------------------------------------------------------------- /assets/cgan.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/cgan.gif -------------------------------------------------------------------------------- /assets/cluster_gan.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/cluster_gan.gif -------------------------------------------------------------------------------- /assets/cogan.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/cogan.gif -------------------------------------------------------------------------------- /assets/context_encoder.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/context_encoder.png -------------------------------------------------------------------------------- /assets/cyclegan.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/cyclegan.png -------------------------------------------------------------------------------- /assets/dcgan.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/dcgan.gif -------------------------------------------------------------------------------- /assets/discogan.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/discogan.png -------------------------------------------------------------------------------- /assets/enhanced_superresgan.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/enhanced_superresgan.png -------------------------------------------------------------------------------- /assets/esrgan.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/esrgan.png -------------------------------------------------------------------------------- /assets/gan.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/gan.gif -------------------------------------------------------------------------------- /assets/infogan.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/infogan.gif -------------------------------------------------------------------------------- /assets/infogan.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/infogan.png -------------------------------------------------------------------------------- /assets/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/logo.png -------------------------------------------------------------------------------- /assets/munit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/munit.png -------------------------------------------------------------------------------- /assets/pix2pix.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/pix2pix.png -------------------------------------------------------------------------------- /assets/pixelda.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/pixelda.png -------------------------------------------------------------------------------- /assets/stargan.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/stargan.png -------------------------------------------------------------------------------- /assets/superresgan.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/superresgan.png -------------------------------------------------------------------------------- /assets/wgan_div.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/wgan_div.gif -------------------------------------------------------------------------------- /assets/wgan_div.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/wgan_div.png -------------------------------------------------------------------------------- /assets/wgan_gp.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/assets/wgan_gp.gif -------------------------------------------------------------------------------- /data/download_cyclegan_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/data/download_cyclegan_dataset.sh -------------------------------------------------------------------------------- /data/download_pix2pix_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/data/download_pix2pix_dataset.sh -------------------------------------------------------------------------------- /implementations/aae/aae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/aae/aae.py -------------------------------------------------------------------------------- /implementations/acgan/acgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/acgan/acgan.py -------------------------------------------------------------------------------- /implementations/began/began.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/began/began.py -------------------------------------------------------------------------------- /implementations/bgan/bgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/bgan/bgan.py -------------------------------------------------------------------------------- /implementations/bicyclegan/bicyclegan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/bicyclegan/bicyclegan.py -------------------------------------------------------------------------------- /implementations/bicyclegan/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/bicyclegan/datasets.py -------------------------------------------------------------------------------- /implementations/bicyclegan/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/bicyclegan/models.py -------------------------------------------------------------------------------- /implementations/ccgan/ccgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/ccgan/ccgan.py -------------------------------------------------------------------------------- /implementations/ccgan/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/ccgan/datasets.py -------------------------------------------------------------------------------- /implementations/ccgan/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/ccgan/models.py -------------------------------------------------------------------------------- /implementations/cgan/cgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/cgan/cgan.py -------------------------------------------------------------------------------- /implementations/cluster_gan/clustergan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/cluster_gan/clustergan.py -------------------------------------------------------------------------------- /implementations/cogan/cogan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/cogan/cogan.py -------------------------------------------------------------------------------- /implementations/cogan/mnistm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/cogan/mnistm.py -------------------------------------------------------------------------------- /implementations/context_encoder/context_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/context_encoder/context_encoder.py -------------------------------------------------------------------------------- /implementations/context_encoder/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/context_encoder/datasets.py -------------------------------------------------------------------------------- /implementations/context_encoder/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/context_encoder/models.py -------------------------------------------------------------------------------- /implementations/cyclegan/cyclegan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/cyclegan/cyclegan.py -------------------------------------------------------------------------------- /implementations/cyclegan/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/cyclegan/datasets.py -------------------------------------------------------------------------------- /implementations/cyclegan/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/cyclegan/models.py -------------------------------------------------------------------------------- /implementations/cyclegan/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/cyclegan/utils.py -------------------------------------------------------------------------------- /implementations/dcgan/dcgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/dcgan/dcgan.py -------------------------------------------------------------------------------- /implementations/discogan/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/discogan/datasets.py -------------------------------------------------------------------------------- /implementations/discogan/discogan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/discogan/discogan.py -------------------------------------------------------------------------------- /implementations/discogan/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/discogan/models.py -------------------------------------------------------------------------------- /implementations/dragan/dragan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/dragan/dragan.py -------------------------------------------------------------------------------- /implementations/dualgan/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/dualgan/datasets.py -------------------------------------------------------------------------------- /implementations/dualgan/dualgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/dualgan/dualgan.py -------------------------------------------------------------------------------- /implementations/dualgan/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/dualgan/models.py -------------------------------------------------------------------------------- /implementations/ebgan/ebgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/ebgan/ebgan.py -------------------------------------------------------------------------------- /implementations/esrgan/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/esrgan/datasets.py -------------------------------------------------------------------------------- /implementations/esrgan/esrgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/esrgan/esrgan.py -------------------------------------------------------------------------------- /implementations/esrgan/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/esrgan/models.py -------------------------------------------------------------------------------- /implementations/esrgan/test_on_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/esrgan/test_on_image.py -------------------------------------------------------------------------------- /implementations/gan/gan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/gan/gan.py -------------------------------------------------------------------------------- /implementations/infogan/infogan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/infogan/infogan.py -------------------------------------------------------------------------------- /implementations/lsgan/lsgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/lsgan/lsgan.py -------------------------------------------------------------------------------- /implementations/munit/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/munit/datasets.py -------------------------------------------------------------------------------- /implementations/munit/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/munit/models.py -------------------------------------------------------------------------------- /implementations/munit/munit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/munit/munit.py -------------------------------------------------------------------------------- /implementations/pix2pix/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/pix2pix/datasets.py -------------------------------------------------------------------------------- /implementations/pix2pix/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/pix2pix/models.py -------------------------------------------------------------------------------- /implementations/pix2pix/pix2pix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/pix2pix/pix2pix.py -------------------------------------------------------------------------------- /implementations/pixelda/mnistm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/pixelda/mnistm.py -------------------------------------------------------------------------------- /implementations/pixelda/pixelda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/pixelda/pixelda.py -------------------------------------------------------------------------------- /implementations/relativistic_gan/relativistic_gan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/relativistic_gan/relativistic_gan.py -------------------------------------------------------------------------------- /implementations/sgan/sgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/sgan/sgan.py -------------------------------------------------------------------------------- /implementations/softmax_gan/softmax_gan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/softmax_gan/softmax_gan.py -------------------------------------------------------------------------------- /implementations/srgan/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/srgan/datasets.py -------------------------------------------------------------------------------- /implementations/srgan/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/srgan/models.py -------------------------------------------------------------------------------- /implementations/srgan/srgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/srgan/srgan.py -------------------------------------------------------------------------------- /implementations/stargan/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/stargan/datasets.py -------------------------------------------------------------------------------- /implementations/stargan/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/stargan/models.py -------------------------------------------------------------------------------- /implementations/stargan/stargan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/stargan/stargan.py -------------------------------------------------------------------------------- /implementations/unit/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/unit/datasets.py -------------------------------------------------------------------------------- /implementations/unit/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/unit/models.py -------------------------------------------------------------------------------- /implementations/unit/unit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/unit/unit.py -------------------------------------------------------------------------------- /implementations/wgan/wgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/wgan/wgan.py -------------------------------------------------------------------------------- /implementations/wgan_div/wgan_div.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/wgan_div/wgan_div.py -------------------------------------------------------------------------------- /implementations/wgan_gp/wgan_gp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/implementations/wgan_gp/wgan_gp.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eriklindernoren/PyTorch-GAN/HEAD/requirements.txt --------------------------------------------------------------------------------