├── README.md ├── projection ├── LICENSE.md ├── README.md ├── __init__.py ├── configs │ ├── sn_projection.yml │ ├── sn_projection_caltech.yml │ ├── sn_projection_cub.yml │ └── sn_projection_flower.yml ├── datasets │ ├── flower_example │ │ ├── 000 │ │ │ └── image_06734.jpg │ │ └── 001 │ │ │ └── image_05087.jpg │ ├── imagenet.py │ ├── make_image_list.py │ └── preprocess.sh ├── dis_models │ ├── resblocks.py │ ├── snresnet.py │ └── snresnet_small.py ├── evaluation.py ├── evaluations │ ├── __init__.py │ ├── calc_inception_score.py │ ├── calc_intra_FID.py │ ├── calc_ref_stats.py │ ├── gen_images.py │ └── gen_interpolated_images.py ├── finetune.py ├── gen_models │ ├── __init__.py │ ├── resblocks.py │ ├── resnet.py │ └── resnet_small.py ├── source │ ├── __init__.py │ ├── functions │ │ ├── __init__.py │ │ └── max_sv.py │ ├── inception │ │ ├── __init__.py │ │ ├── download.py │ │ ├── inception_score.py │ │ └── inception_score_tf.py │ ├── links │ │ ├── __init__.py │ │ ├── categorical_conditional_batch_normalization.py │ │ ├── conditional_batch_normalization.py │ │ ├── sn_convolution_2d.py │ │ ├── sn_convolution_nd.py │ │ ├── sn_embed_id.py │ │ └── sn_linear.py │ ├── miscs │ │ ├── __init__.py │ │ └── random_samples.py │ └── yaml_utils.py ├── train.py └── updater.py ├── resources ├── cat_trend.gif ├── dog_trend.gif ├── projection │ ├── caltech_base.png │ ├── caltech_freeze.png │ ├── cub_base.png │ ├── cub_freeze.png │ ├── flower_base.png │ └── flower_freeze.png └── stylegan │ ├── bear.png │ ├── cat.png │ ├── chicken.png │ ├── cow.png │ ├── deer.png │ ├── dog.png │ ├── duck.png │ ├── eagle.png │ ├── elephant.png │ ├── fate.png │ ├── haruhi.png │ ├── human.png │ ├── lelouch.png │ ├── lion.png │ ├── miku.png │ ├── mio.png │ ├── monkey.png │ ├── mouse.png │ ├── nanoha.png │ ├── original.png │ ├── panda.png │ ├── pig.png │ ├── pigeon.png │ ├── rabbit.png │ ├── reimu.png │ ├── sakura.png │ ├── shana.png │ ├── sheep.png │ ├── tiger.png │ ├── wolf.png │ └── yuki.png └── stylegan ├── LICENSE ├── README.md ├── dataset.py ├── finetune.py ├── generate.py ├── loss ├── AdaBIGGANLoss.py ├── Vgg16PerceptualLoss.py └── __init__.py ├── metric ├── fid_score.py ├── inception.py ├── kid_score.py ├── metric.py └── swd_score.py ├── model.py ├── precompute_acts.py ├── prepare_data.py ├── test.py └── train.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/README.md -------------------------------------------------------------------------------- /projection/LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/LICENSE.md -------------------------------------------------------------------------------- /projection/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/README.md -------------------------------------------------------------------------------- /projection/__init__.py: -------------------------------------------------------------------------------- 1 | #majiunko -------------------------------------------------------------------------------- /projection/configs/sn_projection.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/configs/sn_projection.yml -------------------------------------------------------------------------------- /projection/configs/sn_projection_caltech.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/configs/sn_projection_caltech.yml -------------------------------------------------------------------------------- /projection/configs/sn_projection_cub.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/configs/sn_projection_cub.yml -------------------------------------------------------------------------------- /projection/configs/sn_projection_flower.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/configs/sn_projection_flower.yml -------------------------------------------------------------------------------- /projection/datasets/flower_example/000/image_06734.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/datasets/flower_example/000/image_06734.jpg -------------------------------------------------------------------------------- /projection/datasets/flower_example/001/image_05087.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/datasets/flower_example/001/image_05087.jpg -------------------------------------------------------------------------------- /projection/datasets/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/datasets/imagenet.py -------------------------------------------------------------------------------- /projection/datasets/make_image_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/datasets/make_image_list.py -------------------------------------------------------------------------------- /projection/datasets/preprocess.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/datasets/preprocess.sh -------------------------------------------------------------------------------- /projection/dis_models/resblocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/dis_models/resblocks.py -------------------------------------------------------------------------------- /projection/dis_models/snresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/dis_models/snresnet.py -------------------------------------------------------------------------------- /projection/dis_models/snresnet_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/dis_models/snresnet_small.py -------------------------------------------------------------------------------- /projection/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/evaluation.py -------------------------------------------------------------------------------- /projection/evaluations/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /projection/evaluations/calc_inception_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/evaluations/calc_inception_score.py -------------------------------------------------------------------------------- /projection/evaluations/calc_intra_FID.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/evaluations/calc_intra_FID.py -------------------------------------------------------------------------------- /projection/evaluations/calc_ref_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/evaluations/calc_ref_stats.py -------------------------------------------------------------------------------- /projection/evaluations/gen_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/evaluations/gen_images.py -------------------------------------------------------------------------------- /projection/evaluations/gen_interpolated_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/evaluations/gen_interpolated_images.py -------------------------------------------------------------------------------- /projection/finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/finetune.py -------------------------------------------------------------------------------- /projection/gen_models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /projection/gen_models/resblocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/gen_models/resblocks.py -------------------------------------------------------------------------------- /projection/gen_models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/gen_models/resnet.py -------------------------------------------------------------------------------- /projection/gen_models/resnet_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/gen_models/resnet_small.py -------------------------------------------------------------------------------- /projection/source/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /projection/source/functions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /projection/source/functions/max_sv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/source/functions/max_sv.py -------------------------------------------------------------------------------- /projection/source/inception/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /projection/source/inception/download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/source/inception/download.py -------------------------------------------------------------------------------- /projection/source/inception/inception_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/source/inception/inception_score.py -------------------------------------------------------------------------------- /projection/source/inception/inception_score_tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/source/inception/inception_score_tf.py -------------------------------------------------------------------------------- /projection/source/links/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /projection/source/links/categorical_conditional_batch_normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/source/links/categorical_conditional_batch_normalization.py -------------------------------------------------------------------------------- /projection/source/links/conditional_batch_normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/source/links/conditional_batch_normalization.py -------------------------------------------------------------------------------- /projection/source/links/sn_convolution_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/source/links/sn_convolution_2d.py -------------------------------------------------------------------------------- /projection/source/links/sn_convolution_nd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/source/links/sn_convolution_nd.py -------------------------------------------------------------------------------- /projection/source/links/sn_embed_id.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/source/links/sn_embed_id.py -------------------------------------------------------------------------------- /projection/source/links/sn_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/source/links/sn_linear.py -------------------------------------------------------------------------------- /projection/source/miscs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /projection/source/miscs/random_samples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/source/miscs/random_samples.py -------------------------------------------------------------------------------- /projection/source/yaml_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/source/yaml_utils.py -------------------------------------------------------------------------------- /projection/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/train.py -------------------------------------------------------------------------------- /projection/updater.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/projection/updater.py -------------------------------------------------------------------------------- /resources/cat_trend.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/cat_trend.gif -------------------------------------------------------------------------------- /resources/dog_trend.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/dog_trend.gif -------------------------------------------------------------------------------- /resources/projection/caltech_base.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/projection/caltech_base.png -------------------------------------------------------------------------------- /resources/projection/caltech_freeze.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/projection/caltech_freeze.png -------------------------------------------------------------------------------- /resources/projection/cub_base.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/projection/cub_base.png -------------------------------------------------------------------------------- /resources/projection/cub_freeze.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/projection/cub_freeze.png -------------------------------------------------------------------------------- /resources/projection/flower_base.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/projection/flower_base.png -------------------------------------------------------------------------------- /resources/projection/flower_freeze.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/projection/flower_freeze.png -------------------------------------------------------------------------------- /resources/stylegan/bear.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/bear.png -------------------------------------------------------------------------------- /resources/stylegan/cat.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/cat.png -------------------------------------------------------------------------------- /resources/stylegan/chicken.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/chicken.png -------------------------------------------------------------------------------- /resources/stylegan/cow.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/cow.png -------------------------------------------------------------------------------- /resources/stylegan/deer.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/deer.png -------------------------------------------------------------------------------- /resources/stylegan/dog.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/dog.png -------------------------------------------------------------------------------- /resources/stylegan/duck.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/duck.png -------------------------------------------------------------------------------- /resources/stylegan/eagle.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/eagle.png -------------------------------------------------------------------------------- /resources/stylegan/elephant.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/elephant.png -------------------------------------------------------------------------------- /resources/stylegan/fate.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/fate.png -------------------------------------------------------------------------------- /resources/stylegan/haruhi.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/haruhi.png -------------------------------------------------------------------------------- /resources/stylegan/human.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/human.png -------------------------------------------------------------------------------- /resources/stylegan/lelouch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/lelouch.png -------------------------------------------------------------------------------- /resources/stylegan/lion.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/lion.png -------------------------------------------------------------------------------- /resources/stylegan/miku.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/miku.png -------------------------------------------------------------------------------- /resources/stylegan/mio.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/mio.png -------------------------------------------------------------------------------- /resources/stylegan/monkey.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/monkey.png -------------------------------------------------------------------------------- /resources/stylegan/mouse.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/mouse.png -------------------------------------------------------------------------------- /resources/stylegan/nanoha.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/nanoha.png -------------------------------------------------------------------------------- /resources/stylegan/original.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/original.png -------------------------------------------------------------------------------- /resources/stylegan/panda.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/panda.png -------------------------------------------------------------------------------- /resources/stylegan/pig.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/pig.png -------------------------------------------------------------------------------- /resources/stylegan/pigeon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/pigeon.png -------------------------------------------------------------------------------- /resources/stylegan/rabbit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/rabbit.png -------------------------------------------------------------------------------- /resources/stylegan/reimu.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/reimu.png -------------------------------------------------------------------------------- /resources/stylegan/sakura.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/sakura.png -------------------------------------------------------------------------------- /resources/stylegan/shana.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/shana.png -------------------------------------------------------------------------------- /resources/stylegan/sheep.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/sheep.png -------------------------------------------------------------------------------- /resources/stylegan/tiger.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/tiger.png -------------------------------------------------------------------------------- /resources/stylegan/wolf.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/wolf.png -------------------------------------------------------------------------------- /resources/stylegan/yuki.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/resources/stylegan/yuki.png -------------------------------------------------------------------------------- /stylegan/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/stylegan/LICENSE -------------------------------------------------------------------------------- /stylegan/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/stylegan/README.md -------------------------------------------------------------------------------- /stylegan/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/stylegan/dataset.py -------------------------------------------------------------------------------- /stylegan/finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/stylegan/finetune.py -------------------------------------------------------------------------------- /stylegan/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/stylegan/generate.py -------------------------------------------------------------------------------- /stylegan/loss/AdaBIGGANLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/stylegan/loss/AdaBIGGANLoss.py -------------------------------------------------------------------------------- /stylegan/loss/Vgg16PerceptualLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/stylegan/loss/Vgg16PerceptualLoss.py -------------------------------------------------------------------------------- /stylegan/loss/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /stylegan/metric/fid_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/stylegan/metric/fid_score.py -------------------------------------------------------------------------------- /stylegan/metric/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/stylegan/metric/inception.py -------------------------------------------------------------------------------- /stylegan/metric/kid_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/stylegan/metric/kid_score.py -------------------------------------------------------------------------------- /stylegan/metric/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/stylegan/metric/metric.py -------------------------------------------------------------------------------- /stylegan/metric/swd_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/stylegan/metric/swd_score.py -------------------------------------------------------------------------------- /stylegan/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/stylegan/model.py -------------------------------------------------------------------------------- /stylegan/precompute_acts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/stylegan/precompute_acts.py -------------------------------------------------------------------------------- /stylegan/prepare_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/stylegan/prepare_data.py -------------------------------------------------------------------------------- /stylegan/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/stylegan/test.py -------------------------------------------------------------------------------- /stylegan/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sangwoomo/FreezeD/HEAD/stylegan/train.py --------------------------------------------------------------------------------