├── .gitignore ├── LICENSE ├── README.md ├── configs ├── __init__.py ├── data_configs.py ├── paths_config.py └── transforms_config.py ├── criteria ├── KLDiv_loss.py ├── __init__.py ├── clip_loss.py ├── lpips │ ├── __init__.py │ ├── lpips.py │ ├── networks.py │ └── utils.py └── w_norm.py ├── datasets ├── __init__.py └── images_dataset.py ├── docs ├── GUI.gif └── teaser.png ├── encode_ffhq.ipynb ├── environment.yml ├── launch_training.sh ├── licenses ├── LICENSE_HuangYG123 ├── LICENSE_S-aiueo32 ├── LICENSE_TreB1eN ├── LICENSE_lessw2020 └── LICENSE_rosinality ├── models ├── __init__.py ├── encoders │ ├── __init__.py │ ├── latent_mappers.py │ └── t2s_encoders.py ├── stylegan2 │ ├── __init__.py │ ├── model.py │ └── op │ │ ├── __init__.py │ │ ├── fused_act.py │ │ ├── fused_bias_act.cpp │ │ ├── fused_bias_act_kernel.cu │ │ ├── upfirdn2d.cpp │ │ ├── upfirdn2d.py │ │ └── upfirdn2d_kernel.cu └── t2s.py ├── options ├── __init__.py └── train_options.py ├── scripts └── train.py ├── streamlit_gui ├── README.md ├── app.py └── streamlit_utils.py ├── training ├── __init__.py └── coach.py └── utils ├── __init__.py ├── common.py ├── data_utils.py ├── train_utils.py └── wandb_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/README.md -------------------------------------------------------------------------------- /configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/data_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/configs/data_configs.py -------------------------------------------------------------------------------- /configs/paths_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/configs/paths_config.py -------------------------------------------------------------------------------- /configs/transforms_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/configs/transforms_config.py -------------------------------------------------------------------------------- /criteria/KLDiv_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/criteria/KLDiv_loss.py -------------------------------------------------------------------------------- /criteria/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /criteria/clip_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/criteria/clip_loss.py -------------------------------------------------------------------------------- /criteria/lpips/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /criteria/lpips/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/criteria/lpips/lpips.py -------------------------------------------------------------------------------- /criteria/lpips/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/criteria/lpips/networks.py -------------------------------------------------------------------------------- /criteria/lpips/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/criteria/lpips/utils.py -------------------------------------------------------------------------------- /criteria/w_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/criteria/w_norm.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/images_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/datasets/images_dataset.py -------------------------------------------------------------------------------- /docs/GUI.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/docs/GUI.gif -------------------------------------------------------------------------------- /docs/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/docs/teaser.png -------------------------------------------------------------------------------- /encode_ffhq.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/encode_ffhq.ipynb -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/environment.yml -------------------------------------------------------------------------------- /launch_training.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/launch_training.sh -------------------------------------------------------------------------------- /licenses/LICENSE_HuangYG123: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/licenses/LICENSE_HuangYG123 -------------------------------------------------------------------------------- /licenses/LICENSE_S-aiueo32: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/licenses/LICENSE_S-aiueo32 -------------------------------------------------------------------------------- /licenses/LICENSE_TreB1eN: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/licenses/LICENSE_TreB1eN -------------------------------------------------------------------------------- /licenses/LICENSE_lessw2020: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/licenses/LICENSE_lessw2020 -------------------------------------------------------------------------------- /licenses/LICENSE_rosinality: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/licenses/LICENSE_rosinality -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/encoders/latent_mappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/models/encoders/latent_mappers.py -------------------------------------------------------------------------------- /models/encoders/t2s_encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/models/encoders/t2s_encoders.py -------------------------------------------------------------------------------- /models/stylegan2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/stylegan2/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/models/stylegan2/model.py -------------------------------------------------------------------------------- /models/stylegan2/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/models/stylegan2/op/__init__.py -------------------------------------------------------------------------------- /models/stylegan2/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/models/stylegan2/op/fused_act.py -------------------------------------------------------------------------------- /models/stylegan2/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/models/stylegan2/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /models/stylegan2/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/models/stylegan2/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /models/stylegan2/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/models/stylegan2/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /models/stylegan2/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/models/stylegan2/op/upfirdn2d.py -------------------------------------------------------------------------------- /models/stylegan2/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/models/stylegan2/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /models/t2s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/models/t2s.py -------------------------------------------------------------------------------- /options/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /options/train_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/options/train_options.py -------------------------------------------------------------------------------- /scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/scripts/train.py -------------------------------------------------------------------------------- /streamlit_gui/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/streamlit_gui/README.md -------------------------------------------------------------------------------- /streamlit_gui/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/streamlit_gui/app.py -------------------------------------------------------------------------------- /streamlit_gui/streamlit_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/streamlit_gui/streamlit_utils.py -------------------------------------------------------------------------------- /training/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /training/coach.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/training/coach.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/utils/common.py -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/utils/train_utils.py -------------------------------------------------------------------------------- /utils/wandb_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duxiaodan/Fast_text2StyleGAN/HEAD/utils/wandb_utils.py --------------------------------------------------------------------------------