├── .gitignore ├── README.md ├── configs ├── __init__.py ├── global_config.py ├── hyperparameters.py └── paths_config.py ├── criteria ├── __init__.py ├── l2_loss.py └── localitly_regularizer.py ├── datasets └── image_list_dataset.py ├── dnnlib ├── __init__.py └── util.py ├── download_models.sh ├── edit_video.py ├── edit_video_stitching_tuning.py ├── editings ├── latent_editor.py ├── styleclip_global │ ├── S_mean_std │ └── fs3.npy ├── styleclip_global_utils.py └── w_directions │ ├── age.npy │ ├── eye_distance.npy │ ├── eye_eyebrow_distance.npy │ ├── eye_ratio.npy │ ├── eyes_open.npy │ ├── gender.npy │ ├── lip_ratio.npy │ ├── mouth_open.npy │ ├── mouth_ratio.npy │ ├── nose_mouth_distance.npy │ ├── nose_ratio.npy │ ├── nose_tip.npy │ ├── pitch.npy │ ├── roll.npy │ ├── smile.npy │ └── yaw.npy ├── models ├── __init__.py ├── e4e │ ├── __init__.py │ ├── discriminator.py │ ├── encoders │ │ ├── __init__.py │ │ ├── helpers.py │ │ └── psp_encoders.py │ ├── latent_codes_pool.py │ ├── psp.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 ├── seg_model_2.py └── stylegan2 │ ├── __init__.py │ ├── model.py │ └── op │ ├── __init__.py │ ├── fused_act.py │ └── upfirdn2d.py ├── requirements.txt ├── torch_utils ├── __init__.py ├── custom_ops.py ├── misc.py ├── ops │ ├── __init__.py │ ├── bias_act.cpp │ ├── bias_act.cu │ ├── bias_act.h │ ├── bias_act.py │ ├── conv2d_gradfix.py │ ├── conv2d_resample.py │ ├── fma.py │ ├── grid_sample_gradfix.py │ ├── upfirdn2d.cpp │ ├── upfirdn2d.cu │ ├── upfirdn2d.h │ └── upfirdn2d.py ├── persistence.py └── training_stats.py ├── train.py ├── training ├── coaches │ ├── __init__.py │ └── coach.py ├── networks.py └── projectors │ └── w_projector.py └── utils ├── alignment.py ├── data_utils.py ├── edit_utils.py ├── image_utils.py ├── log_utils.py ├── models_utils.py └── morphology.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/README.md -------------------------------------------------------------------------------- /configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/global_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/configs/global_config.py -------------------------------------------------------------------------------- /configs/hyperparameters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/configs/hyperparameters.py -------------------------------------------------------------------------------- /configs/paths_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/configs/paths_config.py -------------------------------------------------------------------------------- /criteria/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /criteria/l2_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/criteria/l2_loss.py -------------------------------------------------------------------------------- /criteria/localitly_regularizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/criteria/localitly_regularizer.py -------------------------------------------------------------------------------- /datasets/image_list_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/datasets/image_list_dataset.py -------------------------------------------------------------------------------- /dnnlib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/dnnlib/__init__.py -------------------------------------------------------------------------------- /dnnlib/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/dnnlib/util.py -------------------------------------------------------------------------------- /download_models.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/download_models.sh -------------------------------------------------------------------------------- /edit_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/edit_video.py -------------------------------------------------------------------------------- /edit_video_stitching_tuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/edit_video_stitching_tuning.py -------------------------------------------------------------------------------- /editings/latent_editor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/editings/latent_editor.py -------------------------------------------------------------------------------- /editings/styleclip_global/S_mean_std: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/editings/styleclip_global/S_mean_std -------------------------------------------------------------------------------- /editings/styleclip_global/fs3.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/editings/styleclip_global/fs3.npy -------------------------------------------------------------------------------- /editings/styleclip_global_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/editings/styleclip_global_utils.py -------------------------------------------------------------------------------- /editings/w_directions/age.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/editings/w_directions/age.npy -------------------------------------------------------------------------------- /editings/w_directions/eye_distance.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/editings/w_directions/eye_distance.npy -------------------------------------------------------------------------------- /editings/w_directions/eye_eyebrow_distance.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/editings/w_directions/eye_eyebrow_distance.npy -------------------------------------------------------------------------------- /editings/w_directions/eye_ratio.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/editings/w_directions/eye_ratio.npy -------------------------------------------------------------------------------- /editings/w_directions/eyes_open.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/editings/w_directions/eyes_open.npy -------------------------------------------------------------------------------- /editings/w_directions/gender.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/editings/w_directions/gender.npy -------------------------------------------------------------------------------- /editings/w_directions/lip_ratio.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/editings/w_directions/lip_ratio.npy -------------------------------------------------------------------------------- /editings/w_directions/mouth_open.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/editings/w_directions/mouth_open.npy -------------------------------------------------------------------------------- /editings/w_directions/mouth_ratio.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/editings/w_directions/mouth_ratio.npy -------------------------------------------------------------------------------- /editings/w_directions/nose_mouth_distance.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/editings/w_directions/nose_mouth_distance.npy -------------------------------------------------------------------------------- /editings/w_directions/nose_ratio.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/editings/w_directions/nose_ratio.npy -------------------------------------------------------------------------------- /editings/w_directions/nose_tip.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/editings/w_directions/nose_tip.npy -------------------------------------------------------------------------------- /editings/w_directions/pitch.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/editings/w_directions/pitch.npy -------------------------------------------------------------------------------- /editings/w_directions/roll.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/editings/w_directions/roll.npy -------------------------------------------------------------------------------- /editings/w_directions/smile.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/editings/w_directions/smile.npy -------------------------------------------------------------------------------- /editings/w_directions/yaw.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/editings/w_directions/yaw.npy -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/e4e/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/e4e/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/models/e4e/discriminator.py -------------------------------------------------------------------------------- /models/e4e/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/e4e/encoders/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/models/e4e/encoders/helpers.py -------------------------------------------------------------------------------- /models/e4e/encoders/psp_encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/models/e4e/encoders/psp_encoders.py -------------------------------------------------------------------------------- /models/e4e/latent_codes_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/models/e4e/latent_codes_pool.py -------------------------------------------------------------------------------- /models/e4e/psp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/models/e4e/psp.py -------------------------------------------------------------------------------- /models/e4e/stylegan2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/e4e/stylegan2/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/models/e4e/stylegan2/model.py -------------------------------------------------------------------------------- /models/e4e/stylegan2/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/models/e4e/stylegan2/op/__init__.py -------------------------------------------------------------------------------- /models/e4e/stylegan2/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/models/e4e/stylegan2/op/fused_act.py -------------------------------------------------------------------------------- /models/e4e/stylegan2/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/models/e4e/stylegan2/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /models/e4e/stylegan2/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/models/e4e/stylegan2/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /models/e4e/stylegan2/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/models/e4e/stylegan2/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /models/e4e/stylegan2/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/models/e4e/stylegan2/op/upfirdn2d.py -------------------------------------------------------------------------------- /models/e4e/stylegan2/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/models/e4e/stylegan2/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /models/seg_model_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/models/seg_model_2.py -------------------------------------------------------------------------------- /models/stylegan2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/stylegan2/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/models/stylegan2/model.py -------------------------------------------------------------------------------- /models/stylegan2/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/models/stylegan2/op/__init__.py -------------------------------------------------------------------------------- /models/stylegan2/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/models/stylegan2/op/fused_act.py -------------------------------------------------------------------------------- /models/stylegan2/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/models/stylegan2/op/upfirdn2d.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/requirements.txt -------------------------------------------------------------------------------- /torch_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/torch_utils/__init__.py -------------------------------------------------------------------------------- /torch_utils/custom_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/torch_utils/custom_ops.py -------------------------------------------------------------------------------- /torch_utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/torch_utils/misc.py -------------------------------------------------------------------------------- /torch_utils/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/torch_utils/ops/__init__.py -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/torch_utils/ops/bias_act.cpp -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/torch_utils/ops/bias_act.cu -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/torch_utils/ops/bias_act.h -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/torch_utils/ops/bias_act.py -------------------------------------------------------------------------------- /torch_utils/ops/conv2d_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/torch_utils/ops/conv2d_gradfix.py -------------------------------------------------------------------------------- /torch_utils/ops/conv2d_resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/torch_utils/ops/conv2d_resample.py -------------------------------------------------------------------------------- /torch_utils/ops/fma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/torch_utils/ops/fma.py -------------------------------------------------------------------------------- /torch_utils/ops/grid_sample_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/torch_utils/ops/grid_sample_gradfix.py -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/torch_utils/ops/upfirdn2d.cpp -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/torch_utils/ops/upfirdn2d.cu -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/torch_utils/ops/upfirdn2d.h -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/torch_utils/ops/upfirdn2d.py -------------------------------------------------------------------------------- /torch_utils/persistence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/torch_utils/persistence.py -------------------------------------------------------------------------------- /torch_utils/training_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/torch_utils/training_stats.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/train.py -------------------------------------------------------------------------------- /training/coaches/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /training/coaches/coach.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/training/coaches/coach.py -------------------------------------------------------------------------------- /training/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/training/networks.py -------------------------------------------------------------------------------- /training/projectors/w_projector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/training/projectors/w_projector.py -------------------------------------------------------------------------------- /utils/alignment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/utils/alignment.py -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/edit_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/utils/edit_utils.py -------------------------------------------------------------------------------- /utils/image_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/utils/image_utils.py -------------------------------------------------------------------------------- /utils/log_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/utils/log_utils.py -------------------------------------------------------------------------------- /utils/models_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/utils/models_utils.py -------------------------------------------------------------------------------- /utils/morphology.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bycloudai/STIT-Windows/HEAD/utils/morphology.py --------------------------------------------------------------------------------