├── .gitignore ├── LICENSE.txt ├── README.md ├── calc_metrics.py ├── dataset_tool.py ├── dnnlib ├── __init__.py └── util.py ├── environment.yml ├── feature_networks ├── clip │ ├── __init__.py │ ├── bpe_simple_vocab_16e6.txt.gz │ ├── clip.py │ ├── model.py │ └── simple_tokenizer.py ├── constants.py ├── pretrained_builder.py └── vit.py ├── gen_class_samplesheet.py ├── gen_images.py ├── gen_video.py ├── gui_utils ├── __init__.py ├── gl_utils.py ├── glfw_window.py ├── imgui_utils.py ├── imgui_window.py └── text_utils.py ├── in_embeddings └── tf_efficientnet_lite0.pkl ├── legacy.py ├── media ├── banner.png ├── editing_banner.png ├── imagenet_idx2labels.txt ├── inversion.gif ├── jay.png ├── multimodal_truncation.png ├── no_truncation.png ├── system.png ├── teaser.png └── unimodal_truncation.png ├── metrics ├── __init__.py ├── equivariance.py ├── frechet_inception_distance.py ├── inception_score.py ├── kernel_inception_distance.py ├── metric_main.py ├── metric_utils.py ├── perceptual_path_length.py └── precision_recall.py ├── pg_modules ├── blocks.py ├── discriminator.py └── projector.py ├── run_inversion.py ├── run_stylemc.py ├── torch_utils ├── __init__.py ├── custom_ops.py ├── gen_utils.py ├── misc.py ├── ops │ ├── __init__.py │ ├── bias_act.cpp │ ├── bias_act.cu │ ├── bias_act.h │ ├── bias_act.py │ ├── conv2d_gradfix.py │ ├── conv2d_resample.py │ ├── filtered_lrelu.cpp │ ├── filtered_lrelu.cu │ ├── filtered_lrelu.h │ ├── filtered_lrelu.py │ ├── filtered_lrelu_ns.cu │ ├── filtered_lrelu_rd.cu │ ├── filtered_lrelu_wr.cu │ ├── fma.py │ ├── grid_sample_gradfix.py │ ├── upfirdn2d.cpp │ ├── upfirdn2d.cu │ ├── upfirdn2d.h │ └── upfirdn2d.py ├── persistence.py ├── training_stats.py └── utils_spectrum.py ├── train.py ├── training ├── __init__.py ├── augment.py ├── dataset.py ├── diffaug.py ├── loss.py ├── networks_fastgan.py ├── networks_stylegan2.py ├── networks_stylegan3.py ├── networks_stylegan3_resetting.py └── training_loop.py ├── visualizer.py └── viz ├── __init__.py ├── capture_widget.py ├── equivariance_widget.py ├── latent_widget.py ├── layer_widget.py ├── performance_widget.py ├── pickle_widget.py ├── renderer.py ├── stylemix_widget.py └── trunc_noise_widget.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/README.md -------------------------------------------------------------------------------- /calc_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/calc_metrics.py -------------------------------------------------------------------------------- /dataset_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/dataset_tool.py -------------------------------------------------------------------------------- /dnnlib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/dnnlib/__init__.py -------------------------------------------------------------------------------- /dnnlib/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/dnnlib/util.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/environment.yml -------------------------------------------------------------------------------- /feature_networks/clip/__init__.py: -------------------------------------------------------------------------------- 1 | from .clip import * 2 | -------------------------------------------------------------------------------- /feature_networks/clip/bpe_simple_vocab_16e6.txt.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/feature_networks/clip/bpe_simple_vocab_16e6.txt.gz -------------------------------------------------------------------------------- /feature_networks/clip/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/feature_networks/clip/clip.py -------------------------------------------------------------------------------- /feature_networks/clip/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/feature_networks/clip/model.py -------------------------------------------------------------------------------- /feature_networks/clip/simple_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/feature_networks/clip/simple_tokenizer.py -------------------------------------------------------------------------------- /feature_networks/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/feature_networks/constants.py -------------------------------------------------------------------------------- /feature_networks/pretrained_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/feature_networks/pretrained_builder.py -------------------------------------------------------------------------------- /feature_networks/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/feature_networks/vit.py -------------------------------------------------------------------------------- /gen_class_samplesheet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/gen_class_samplesheet.py -------------------------------------------------------------------------------- /gen_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/gen_images.py -------------------------------------------------------------------------------- /gen_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/gen_video.py -------------------------------------------------------------------------------- /gui_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/gui_utils/__init__.py -------------------------------------------------------------------------------- /gui_utils/gl_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/gui_utils/gl_utils.py -------------------------------------------------------------------------------- /gui_utils/glfw_window.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/gui_utils/glfw_window.py -------------------------------------------------------------------------------- /gui_utils/imgui_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/gui_utils/imgui_utils.py -------------------------------------------------------------------------------- /gui_utils/imgui_window.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/gui_utils/imgui_window.py -------------------------------------------------------------------------------- /gui_utils/text_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/gui_utils/text_utils.py -------------------------------------------------------------------------------- /in_embeddings/tf_efficientnet_lite0.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/in_embeddings/tf_efficientnet_lite0.pkl -------------------------------------------------------------------------------- /legacy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/legacy.py -------------------------------------------------------------------------------- /media/banner.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/media/banner.png -------------------------------------------------------------------------------- /media/editing_banner.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/media/editing_banner.png -------------------------------------------------------------------------------- /media/imagenet_idx2labels.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/media/imagenet_idx2labels.txt -------------------------------------------------------------------------------- /media/inversion.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/media/inversion.gif -------------------------------------------------------------------------------- /media/jay.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/media/jay.png -------------------------------------------------------------------------------- /media/multimodal_truncation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/media/multimodal_truncation.png -------------------------------------------------------------------------------- /media/no_truncation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/media/no_truncation.png -------------------------------------------------------------------------------- /media/system.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/media/system.png -------------------------------------------------------------------------------- /media/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/media/teaser.png -------------------------------------------------------------------------------- /media/unimodal_truncation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/media/unimodal_truncation.png -------------------------------------------------------------------------------- /metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/metrics/__init__.py -------------------------------------------------------------------------------- /metrics/equivariance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/metrics/equivariance.py -------------------------------------------------------------------------------- /metrics/frechet_inception_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/metrics/frechet_inception_distance.py -------------------------------------------------------------------------------- /metrics/inception_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/metrics/inception_score.py -------------------------------------------------------------------------------- /metrics/kernel_inception_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/metrics/kernel_inception_distance.py -------------------------------------------------------------------------------- /metrics/metric_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/metrics/metric_main.py -------------------------------------------------------------------------------- /metrics/metric_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/metrics/metric_utils.py -------------------------------------------------------------------------------- /metrics/perceptual_path_length.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/metrics/perceptual_path_length.py -------------------------------------------------------------------------------- /metrics/precision_recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/metrics/precision_recall.py -------------------------------------------------------------------------------- /pg_modules/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/pg_modules/blocks.py -------------------------------------------------------------------------------- /pg_modules/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/pg_modules/discriminator.py -------------------------------------------------------------------------------- /pg_modules/projector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/pg_modules/projector.py -------------------------------------------------------------------------------- /run_inversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/run_inversion.py -------------------------------------------------------------------------------- /run_stylemc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/run_stylemc.py -------------------------------------------------------------------------------- /torch_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/__init__.py -------------------------------------------------------------------------------- /torch_utils/custom_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/custom_ops.py -------------------------------------------------------------------------------- /torch_utils/gen_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/gen_utils.py -------------------------------------------------------------------------------- /torch_utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/misc.py -------------------------------------------------------------------------------- /torch_utils/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/ops/__init__.py -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/ops/bias_act.cpp -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/ops/bias_act.cu -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/ops/bias_act.h -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/ops/bias_act.py -------------------------------------------------------------------------------- /torch_utils/ops/conv2d_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/ops/conv2d_gradfix.py -------------------------------------------------------------------------------- /torch_utils/ops/conv2d_resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/ops/conv2d_resample.py -------------------------------------------------------------------------------- /torch_utils/ops/filtered_lrelu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/ops/filtered_lrelu.cpp -------------------------------------------------------------------------------- /torch_utils/ops/filtered_lrelu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/ops/filtered_lrelu.cu -------------------------------------------------------------------------------- /torch_utils/ops/filtered_lrelu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/ops/filtered_lrelu.h -------------------------------------------------------------------------------- /torch_utils/ops/filtered_lrelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/ops/filtered_lrelu.py -------------------------------------------------------------------------------- /torch_utils/ops/filtered_lrelu_ns.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/ops/filtered_lrelu_ns.cu -------------------------------------------------------------------------------- /torch_utils/ops/filtered_lrelu_rd.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/ops/filtered_lrelu_rd.cu -------------------------------------------------------------------------------- /torch_utils/ops/filtered_lrelu_wr.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/ops/filtered_lrelu_wr.cu -------------------------------------------------------------------------------- /torch_utils/ops/fma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/ops/fma.py -------------------------------------------------------------------------------- /torch_utils/ops/grid_sample_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/ops/grid_sample_gradfix.py -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/ops/upfirdn2d.cpp -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/ops/upfirdn2d.cu -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/ops/upfirdn2d.h -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/ops/upfirdn2d.py -------------------------------------------------------------------------------- /torch_utils/persistence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/persistence.py -------------------------------------------------------------------------------- /torch_utils/training_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/training_stats.py -------------------------------------------------------------------------------- /torch_utils/utils_spectrum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/torch_utils/utils_spectrum.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/train.py -------------------------------------------------------------------------------- /training/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/training/__init__.py -------------------------------------------------------------------------------- /training/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/training/augment.py -------------------------------------------------------------------------------- /training/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/training/dataset.py -------------------------------------------------------------------------------- /training/diffaug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/training/diffaug.py -------------------------------------------------------------------------------- /training/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/training/loss.py -------------------------------------------------------------------------------- /training/networks_fastgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/training/networks_fastgan.py -------------------------------------------------------------------------------- /training/networks_stylegan2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/training/networks_stylegan2.py -------------------------------------------------------------------------------- /training/networks_stylegan3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/training/networks_stylegan3.py -------------------------------------------------------------------------------- /training/networks_stylegan3_resetting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/training/networks_stylegan3_resetting.py -------------------------------------------------------------------------------- /training/training_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/training/training_loop.py -------------------------------------------------------------------------------- /visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/visualizer.py -------------------------------------------------------------------------------- /viz/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/viz/__init__.py -------------------------------------------------------------------------------- /viz/capture_widget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/viz/capture_widget.py -------------------------------------------------------------------------------- /viz/equivariance_widget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/viz/equivariance_widget.py -------------------------------------------------------------------------------- /viz/latent_widget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/viz/latent_widget.py -------------------------------------------------------------------------------- /viz/layer_widget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/viz/layer_widget.py -------------------------------------------------------------------------------- /viz/performance_widget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/viz/performance_widget.py -------------------------------------------------------------------------------- /viz/pickle_widget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/viz/pickle_widget.py -------------------------------------------------------------------------------- /viz/renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/viz/renderer.py -------------------------------------------------------------------------------- /viz/stylemix_widget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/viz/stylemix_widget.py -------------------------------------------------------------------------------- /viz/trunc_noise_widget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/autonomousvision/stylegan-xl/HEAD/viz/trunc_noise_widget.py --------------------------------------------------------------------------------