├── .gitignore ├── EXPERIMENTS.md ├── LICENSE ├── README.md ├── augment ├── __init__.py ├── color_jitter.py ├── spatial.py └── utils.py ├── configs ├── defaults │ ├── augment.gin │ └── gan.gin └── gan │ ├── celeba128 │ ├── celeba128.gin │ └── celeba128_hinge.gin │ ├── cifar10 │ ├── c10_b128.gin │ ├── c10_b256.gin │ ├── c10_b512.gin │ └── c10_b64.gin │ ├── cifar100 │ ├── c100_b128.gin │ ├── c100_b256.gin │ ├── c100_b512.gin │ └── c100_b64.gin │ ├── diffaug │ ├── c100_diffaug.gin │ └── c10_diffaug.gin │ └── stylegan2 │ ├── afhq_cat_style64.gin │ ├── afhq_dog_style64.gin │ ├── afhq_wild_style64.gin │ ├── c100_style32.gin │ ├── c100_style64.gin │ ├── c10_style32.gin │ └── c10_style64.gin ├── datasets.py ├── environment.yml ├── environment_pip.txt ├── evaluate ├── __init__.py ├── classifier.py └── gan.py ├── models ├── __init__.py └── gan │ ├── __init__.py │ ├── base.py │ ├── sndcgan.py │ ├── snresnet.py │ └── stylegan2 │ ├── __init__.py │ ├── discriminator.py │ ├── generator.py │ ├── layers.py │ └── op │ ├── LICENSE │ ├── __init__.py │ ├── fused_act.py │ ├── fused_bias_act.cpp │ ├── fused_bias_act_kernel.cu │ ├── upfirdn2d.cpp │ ├── upfirdn2d.py │ └── upfirdn2d_kernel.cu ├── penalty.py ├── resources ├── c100_real.png ├── c10_real.png ├── cddls │ └── cifar10 │ │ ├── 0.png │ │ ├── 1.png │ │ ├── 2.png │ │ ├── 3.png │ │ ├── 4.png │ │ ├── 5.png │ │ ├── 6.png │ │ ├── 7.png │ │ ├── 8.png │ │ └── 9.png ├── concept.jpg └── demo.jpg ├── test_gan_sample.py ├── test_gan_sample_cddls.py ├── test_lineval.py ├── test_tf_inception.py ├── third_party ├── __init__.py ├── diffaug.py ├── fid │ ├── LICENSE │ ├── __init__.py │ ├── fid_score.py │ └── inception.py ├── gather_layer.py ├── inception_score.py ├── preprocess_celeba_hq.py └── tf │ ├── LICENSE │ ├── README.md │ ├── __init__.py │ ├── examples │ ├── __init__.py │ ├── precalc_stats_cifar10.py │ └── precalc_stats_images.py │ ├── fid.py │ └── inception_score.py ├── train_gan.py ├── train_stylegan2.py ├── train_stylegan2_contraD.py ├── training ├── __init__.py ├── criterion.py └── gan │ ├── __init__.py │ ├── aug.py │ ├── aug_both.py │ ├── contrad.py │ ├── simclr_only.py │ └── std.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/.gitignore -------------------------------------------------------------------------------- /EXPERIMENTS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/EXPERIMENTS.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/README.md -------------------------------------------------------------------------------- /augment/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/augment/__init__.py -------------------------------------------------------------------------------- /augment/color_jitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/augment/color_jitter.py -------------------------------------------------------------------------------- /augment/spatial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/augment/spatial.py -------------------------------------------------------------------------------- /augment/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/augment/utils.py -------------------------------------------------------------------------------- /configs/defaults/augment.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/configs/defaults/augment.gin -------------------------------------------------------------------------------- /configs/defaults/gan.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/configs/defaults/gan.gin -------------------------------------------------------------------------------- /configs/gan/celeba128/celeba128.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/configs/gan/celeba128/celeba128.gin -------------------------------------------------------------------------------- /configs/gan/celeba128/celeba128_hinge.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/configs/gan/celeba128/celeba128_hinge.gin -------------------------------------------------------------------------------- /configs/gan/cifar10/c10_b128.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/configs/gan/cifar10/c10_b128.gin -------------------------------------------------------------------------------- /configs/gan/cifar10/c10_b256.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/configs/gan/cifar10/c10_b256.gin -------------------------------------------------------------------------------- /configs/gan/cifar10/c10_b512.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/configs/gan/cifar10/c10_b512.gin -------------------------------------------------------------------------------- /configs/gan/cifar10/c10_b64.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/configs/gan/cifar10/c10_b64.gin -------------------------------------------------------------------------------- /configs/gan/cifar100/c100_b128.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/configs/gan/cifar100/c100_b128.gin -------------------------------------------------------------------------------- /configs/gan/cifar100/c100_b256.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/configs/gan/cifar100/c100_b256.gin -------------------------------------------------------------------------------- /configs/gan/cifar100/c100_b512.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/configs/gan/cifar100/c100_b512.gin -------------------------------------------------------------------------------- /configs/gan/cifar100/c100_b64.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/configs/gan/cifar100/c100_b64.gin -------------------------------------------------------------------------------- /configs/gan/diffaug/c100_diffaug.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/configs/gan/diffaug/c100_diffaug.gin -------------------------------------------------------------------------------- /configs/gan/diffaug/c10_diffaug.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/configs/gan/diffaug/c10_diffaug.gin -------------------------------------------------------------------------------- /configs/gan/stylegan2/afhq_cat_style64.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/configs/gan/stylegan2/afhq_cat_style64.gin -------------------------------------------------------------------------------- /configs/gan/stylegan2/afhq_dog_style64.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/configs/gan/stylegan2/afhq_dog_style64.gin -------------------------------------------------------------------------------- /configs/gan/stylegan2/afhq_wild_style64.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/configs/gan/stylegan2/afhq_wild_style64.gin -------------------------------------------------------------------------------- /configs/gan/stylegan2/c100_style32.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/configs/gan/stylegan2/c100_style32.gin -------------------------------------------------------------------------------- /configs/gan/stylegan2/c100_style64.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/configs/gan/stylegan2/c100_style64.gin -------------------------------------------------------------------------------- /configs/gan/stylegan2/c10_style32.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/configs/gan/stylegan2/c10_style32.gin -------------------------------------------------------------------------------- /configs/gan/stylegan2/c10_style64.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/configs/gan/stylegan2/c10_style64.gin -------------------------------------------------------------------------------- /datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/datasets.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/environment.yml -------------------------------------------------------------------------------- /environment_pip.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/environment_pip.txt -------------------------------------------------------------------------------- /evaluate/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/evaluate/__init__.py -------------------------------------------------------------------------------- /evaluate/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/evaluate/classifier.py -------------------------------------------------------------------------------- /evaluate/gan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/evaluate/gan.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/gan/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/models/gan/__init__.py -------------------------------------------------------------------------------- /models/gan/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/models/gan/base.py -------------------------------------------------------------------------------- /models/gan/sndcgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/models/gan/sndcgan.py -------------------------------------------------------------------------------- /models/gan/snresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/models/gan/snresnet.py -------------------------------------------------------------------------------- /models/gan/stylegan2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/models/gan/stylegan2/__init__.py -------------------------------------------------------------------------------- /models/gan/stylegan2/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/models/gan/stylegan2/discriminator.py -------------------------------------------------------------------------------- /models/gan/stylegan2/generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/models/gan/stylegan2/generator.py -------------------------------------------------------------------------------- /models/gan/stylegan2/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/models/gan/stylegan2/layers.py -------------------------------------------------------------------------------- /models/gan/stylegan2/op/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/models/gan/stylegan2/op/LICENSE -------------------------------------------------------------------------------- /models/gan/stylegan2/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/models/gan/stylegan2/op/__init__.py -------------------------------------------------------------------------------- /models/gan/stylegan2/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/models/gan/stylegan2/op/fused_act.py -------------------------------------------------------------------------------- /models/gan/stylegan2/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/models/gan/stylegan2/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /models/gan/stylegan2/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/models/gan/stylegan2/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /models/gan/stylegan2/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/models/gan/stylegan2/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /models/gan/stylegan2/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/models/gan/stylegan2/op/upfirdn2d.py -------------------------------------------------------------------------------- /models/gan/stylegan2/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/models/gan/stylegan2/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /penalty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/penalty.py -------------------------------------------------------------------------------- /resources/c100_real.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/resources/c100_real.png -------------------------------------------------------------------------------- /resources/c10_real.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/resources/c10_real.png -------------------------------------------------------------------------------- /resources/cddls/cifar10/0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/resources/cddls/cifar10/0.png -------------------------------------------------------------------------------- /resources/cddls/cifar10/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/resources/cddls/cifar10/1.png -------------------------------------------------------------------------------- /resources/cddls/cifar10/2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/resources/cddls/cifar10/2.png -------------------------------------------------------------------------------- /resources/cddls/cifar10/3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/resources/cddls/cifar10/3.png -------------------------------------------------------------------------------- /resources/cddls/cifar10/4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/resources/cddls/cifar10/4.png -------------------------------------------------------------------------------- /resources/cddls/cifar10/5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/resources/cddls/cifar10/5.png -------------------------------------------------------------------------------- /resources/cddls/cifar10/6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/resources/cddls/cifar10/6.png -------------------------------------------------------------------------------- /resources/cddls/cifar10/7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/resources/cddls/cifar10/7.png -------------------------------------------------------------------------------- /resources/cddls/cifar10/8.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/resources/cddls/cifar10/8.png -------------------------------------------------------------------------------- /resources/cddls/cifar10/9.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/resources/cddls/cifar10/9.png -------------------------------------------------------------------------------- /resources/concept.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/resources/concept.jpg -------------------------------------------------------------------------------- /resources/demo.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/resources/demo.jpg -------------------------------------------------------------------------------- /test_gan_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/test_gan_sample.py -------------------------------------------------------------------------------- /test_gan_sample_cddls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/test_gan_sample_cddls.py -------------------------------------------------------------------------------- /test_lineval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/test_lineval.py -------------------------------------------------------------------------------- /test_tf_inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/test_tf_inception.py -------------------------------------------------------------------------------- /third_party/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/diffaug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/third_party/diffaug.py -------------------------------------------------------------------------------- /third_party/fid/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/third_party/fid/LICENSE -------------------------------------------------------------------------------- /third_party/fid/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/third_party/fid/__init__.py -------------------------------------------------------------------------------- /third_party/fid/fid_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/third_party/fid/fid_score.py -------------------------------------------------------------------------------- /third_party/fid/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/third_party/fid/inception.py -------------------------------------------------------------------------------- /third_party/gather_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/third_party/gather_layer.py -------------------------------------------------------------------------------- /third_party/inception_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/third_party/inception_score.py -------------------------------------------------------------------------------- /third_party/preprocess_celeba_hq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/third_party/preprocess_celeba_hq.py -------------------------------------------------------------------------------- /third_party/tf/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/third_party/tf/LICENSE -------------------------------------------------------------------------------- /third_party/tf/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/third_party/tf/README.md -------------------------------------------------------------------------------- /third_party/tf/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/third_party/tf/__init__.py -------------------------------------------------------------------------------- /third_party/tf/examples/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/tf/examples/precalc_stats_cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/third_party/tf/examples/precalc_stats_cifar10.py -------------------------------------------------------------------------------- /third_party/tf/examples/precalc_stats_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/third_party/tf/examples/precalc_stats_images.py -------------------------------------------------------------------------------- /third_party/tf/fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/third_party/tf/fid.py -------------------------------------------------------------------------------- /third_party/tf/inception_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/third_party/tf/inception_score.py -------------------------------------------------------------------------------- /train_gan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/train_gan.py -------------------------------------------------------------------------------- /train_stylegan2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/train_stylegan2.py -------------------------------------------------------------------------------- /train_stylegan2_contraD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/train_stylegan2_contraD.py -------------------------------------------------------------------------------- /training/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /training/criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/training/criterion.py -------------------------------------------------------------------------------- /training/gan/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/training/gan/__init__.py -------------------------------------------------------------------------------- /training/gan/aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/training/gan/aug.py -------------------------------------------------------------------------------- /training/gan/aug_both.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/training/gan/aug_both.py -------------------------------------------------------------------------------- /training/gan/contrad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/training/gan/contrad.py -------------------------------------------------------------------------------- /training/gan/simclr_only.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/training/gan/simclr_only.py -------------------------------------------------------------------------------- /training/gan/std.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/training/gan/std.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jh-jeong/ContraD/HEAD/utils.py --------------------------------------------------------------------------------