├── LICENSE.md ├── README.md ├── app_gradio.py ├── configs ├── __init__.py ├── data_configs.py ├── dataset_config.yml ├── paths_config.py └── transforms_config.py ├── criteria ├── __init__.py ├── id_loss.py ├── lpips │ ├── __init__.py │ ├── lpips.py │ ├── networks.py │ └── utils.py ├── moco_loss.py └── w_norm.py ├── data ├── 234_sketch.jpg ├── 390.mp4 ├── 529_2.mp4 ├── 540.jpg ├── 684.mp4 ├── ILip77SbmOE.png ├── ILip77SbmOE_45x45.png ├── ILip77SbmOE_mask.png ├── pexels-anthony-shkraba-production-8136210.mp4 └── pexels-daniel-xavier-1239291.jpg ├── datasets ├── __init__.py ├── augmentations.py ├── ffhq_degradation_dataset.py ├── gt_res_dataset.py ├── images_dataset.py └── inference_dataset.py ├── image_translation.py ├── inference_playground.ipynb ├── inversion.py ├── latent_optimization.py ├── models ├── __init__.py ├── bisenet │ ├── LICENSE │ ├── README.md │ ├── model.py │ └── resnet.py ├── encoders │ ├── __init__.py │ ├── helpers.py │ ├── model_irse.py │ └── psp_encoders.py ├── mtcnn │ ├── __init__.py │ ├── mtcnn.py │ └── mtcnn_pytorch │ │ ├── __init__.py │ │ └── src │ │ ├── __init__.py │ │ ├── align_trans.py │ │ ├── box_utils.py │ │ ├── detector.py │ │ ├── first_stage.py │ │ ├── get_nets.py │ │ ├── matlab_cp2tform.py │ │ ├── visualization_utils.py │ │ └── weights │ │ ├── onet.npy │ │ ├── pnet.npy │ │ └── rnet.npy ├── psp.py └── stylegan2 │ ├── __init__.py │ ├── lpips │ ├── __init__.py │ ├── base_model.py │ ├── dist_model.py │ ├── networks_basic.py │ ├── pretrained_networks.py │ └── weights │ │ ├── v0.0 │ │ ├── alex.pth │ │ ├── squeeze.pth │ │ └── vgg.pth │ │ └── v0.1 │ │ ├── alex.pth │ │ ├── squeeze.pth │ │ └── vgg.pth │ ├── model.py │ ├── op │ ├── __init__.py │ ├── conv2d_gradfix.py │ ├── fused_act.py │ ├── readme.md │ └── upfirdn2d.py │ ├── op2 │ ├── __init__.py │ ├── upfirdn2d.cpp │ ├── upfirdn2d.py │ └── upfirdn2d_kernel.cu │ ├── op_old │ ├── __init__.py │ ├── fused_act.py │ ├── fused_bias_act.cpp │ ├── fused_bias_act_kernel.cu │ ├── upfirdn2d.cpp │ ├── upfirdn2d.py │ └── upfirdn2d_kernel.cu │ └── simple_augment.py ├── options ├── __init__.py ├── test_options.py └── train_options.py ├── output └── ILip77SbmOE_inversion.pt ├── pretrained_models └── readme.md ├── scripts ├── align_all_parallel.py ├── calc_id_loss_parallel.py ├── calc_losses_on_images.py ├── download_ffhq1280.py ├── generate_sketch_data.py ├── inference.py ├── pretrain.py ├── style_mixing.py └── train.py ├── training ├── __init__.py ├── coach.py └── ranger.py ├── utils ├── __init__.py ├── common.py ├── data_utils.py ├── inference_utils.py ├── train_utils.py └── wandb_utils.py ├── video_editing.py └── webUI ├── app_task.py └── styleganex_model.py /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/README.md -------------------------------------------------------------------------------- /app_gradio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/app_gradio.py -------------------------------------------------------------------------------- /configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/data_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/configs/data_configs.py -------------------------------------------------------------------------------- /configs/dataset_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/configs/dataset_config.yml -------------------------------------------------------------------------------- /configs/paths_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/configs/paths_config.py -------------------------------------------------------------------------------- /configs/transforms_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/configs/transforms_config.py -------------------------------------------------------------------------------- /criteria/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /criteria/id_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/criteria/id_loss.py -------------------------------------------------------------------------------- /criteria/lpips/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /criteria/lpips/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/criteria/lpips/lpips.py -------------------------------------------------------------------------------- /criteria/lpips/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/criteria/lpips/networks.py -------------------------------------------------------------------------------- /criteria/lpips/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/criteria/lpips/utils.py -------------------------------------------------------------------------------- /criteria/moco_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/criteria/moco_loss.py -------------------------------------------------------------------------------- /criteria/w_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/criteria/w_norm.py -------------------------------------------------------------------------------- /data/234_sketch.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/data/234_sketch.jpg -------------------------------------------------------------------------------- /data/390.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/data/390.mp4 -------------------------------------------------------------------------------- /data/529_2.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/data/529_2.mp4 -------------------------------------------------------------------------------- /data/540.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/data/540.jpg -------------------------------------------------------------------------------- /data/684.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/data/684.mp4 -------------------------------------------------------------------------------- /data/ILip77SbmOE.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/data/ILip77SbmOE.png -------------------------------------------------------------------------------- /data/ILip77SbmOE_45x45.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/data/ILip77SbmOE_45x45.png -------------------------------------------------------------------------------- /data/ILip77SbmOE_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/data/ILip77SbmOE_mask.png -------------------------------------------------------------------------------- /data/pexels-anthony-shkraba-production-8136210.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/data/pexels-anthony-shkraba-production-8136210.mp4 -------------------------------------------------------------------------------- /data/pexels-daniel-xavier-1239291.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/data/pexels-daniel-xavier-1239291.jpg -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/datasets/augmentations.py -------------------------------------------------------------------------------- /datasets/ffhq_degradation_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/datasets/ffhq_degradation_dataset.py -------------------------------------------------------------------------------- /datasets/gt_res_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/datasets/gt_res_dataset.py -------------------------------------------------------------------------------- /datasets/images_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/datasets/images_dataset.py -------------------------------------------------------------------------------- /datasets/inference_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/datasets/inference_dataset.py -------------------------------------------------------------------------------- /image_translation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/image_translation.py -------------------------------------------------------------------------------- /inference_playground.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/inference_playground.ipynb -------------------------------------------------------------------------------- /inversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/inversion.py -------------------------------------------------------------------------------- /latent_optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/latent_optimization.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/bisenet/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/bisenet/LICENSE -------------------------------------------------------------------------------- /models/bisenet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/bisenet/README.md -------------------------------------------------------------------------------- /models/bisenet/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/bisenet/model.py -------------------------------------------------------------------------------- /models/bisenet/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/bisenet/resnet.py -------------------------------------------------------------------------------- /models/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/encoders/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/encoders/helpers.py -------------------------------------------------------------------------------- /models/encoders/model_irse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/encoders/model_irse.py -------------------------------------------------------------------------------- /models/encoders/psp_encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/encoders/psp_encoders.py -------------------------------------------------------------------------------- /models/mtcnn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/mtcnn/mtcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/mtcnn/mtcnn.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/mtcnn/mtcnn_pytorch/src/__init__.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/align_trans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/mtcnn/mtcnn_pytorch/src/align_trans.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/box_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/mtcnn/mtcnn_pytorch/src/box_utils.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/mtcnn/mtcnn_pytorch/src/detector.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/first_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/mtcnn/mtcnn_pytorch/src/first_stage.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/get_nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/mtcnn/mtcnn_pytorch/src/get_nets.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/matlab_cp2tform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/mtcnn/mtcnn_pytorch/src/matlab_cp2tform.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/visualization_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/mtcnn/mtcnn_pytorch/src/visualization_utils.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/weights/onet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/mtcnn/mtcnn_pytorch/src/weights/onet.npy -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/weights/pnet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/mtcnn/mtcnn_pytorch/src/weights/pnet.npy -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/weights/rnet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/mtcnn/mtcnn_pytorch/src/weights/rnet.npy -------------------------------------------------------------------------------- /models/psp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/psp.py -------------------------------------------------------------------------------- /models/stylegan2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/stylegan2/lpips/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/lpips/__init__.py -------------------------------------------------------------------------------- /models/stylegan2/lpips/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/lpips/base_model.py -------------------------------------------------------------------------------- /models/stylegan2/lpips/dist_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/lpips/dist_model.py -------------------------------------------------------------------------------- /models/stylegan2/lpips/networks_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/lpips/networks_basic.py -------------------------------------------------------------------------------- /models/stylegan2/lpips/pretrained_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/lpips/pretrained_networks.py -------------------------------------------------------------------------------- /models/stylegan2/lpips/weights/v0.0/alex.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/lpips/weights/v0.0/alex.pth -------------------------------------------------------------------------------- /models/stylegan2/lpips/weights/v0.0/squeeze.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/lpips/weights/v0.0/squeeze.pth -------------------------------------------------------------------------------- /models/stylegan2/lpips/weights/v0.0/vgg.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/lpips/weights/v0.0/vgg.pth -------------------------------------------------------------------------------- /models/stylegan2/lpips/weights/v0.1/alex.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/lpips/weights/v0.1/alex.pth -------------------------------------------------------------------------------- /models/stylegan2/lpips/weights/v0.1/squeeze.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/lpips/weights/v0.1/squeeze.pth -------------------------------------------------------------------------------- /models/stylegan2/lpips/weights/v0.1/vgg.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/lpips/weights/v0.1/vgg.pth -------------------------------------------------------------------------------- /models/stylegan2/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/model.py -------------------------------------------------------------------------------- /models/stylegan2/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/op/__init__.py -------------------------------------------------------------------------------- /models/stylegan2/op/conv2d_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/op/conv2d_gradfix.py -------------------------------------------------------------------------------- /models/stylegan2/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/op/fused_act.py -------------------------------------------------------------------------------- /models/stylegan2/op/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/op/readme.md -------------------------------------------------------------------------------- /models/stylegan2/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/op/upfirdn2d.py -------------------------------------------------------------------------------- /models/stylegan2/op2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/op2/__init__.py -------------------------------------------------------------------------------- /models/stylegan2/op2/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/op2/upfirdn2d.cpp -------------------------------------------------------------------------------- /models/stylegan2/op2/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/op2/upfirdn2d.py -------------------------------------------------------------------------------- /models/stylegan2/op2/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/op2/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /models/stylegan2/op_old/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/op_old/__init__.py -------------------------------------------------------------------------------- /models/stylegan2/op_old/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/op_old/fused_act.py -------------------------------------------------------------------------------- /models/stylegan2/op_old/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/op_old/fused_bias_act.cpp -------------------------------------------------------------------------------- /models/stylegan2/op_old/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/op_old/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /models/stylegan2/op_old/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/op_old/upfirdn2d.cpp -------------------------------------------------------------------------------- /models/stylegan2/op_old/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/op_old/upfirdn2d.py -------------------------------------------------------------------------------- /models/stylegan2/op_old/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/op_old/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /models/stylegan2/simple_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/models/stylegan2/simple_augment.py -------------------------------------------------------------------------------- /options/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /options/test_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/options/test_options.py -------------------------------------------------------------------------------- /options/train_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/options/train_options.py -------------------------------------------------------------------------------- /output/ILip77SbmOE_inversion.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/output/ILip77SbmOE_inversion.pt -------------------------------------------------------------------------------- /pretrained_models/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/pretrained_models/readme.md -------------------------------------------------------------------------------- /scripts/align_all_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/scripts/align_all_parallel.py -------------------------------------------------------------------------------- /scripts/calc_id_loss_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/scripts/calc_id_loss_parallel.py -------------------------------------------------------------------------------- /scripts/calc_losses_on_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/scripts/calc_losses_on_images.py -------------------------------------------------------------------------------- /scripts/download_ffhq1280.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/scripts/download_ffhq1280.py -------------------------------------------------------------------------------- /scripts/generate_sketch_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/scripts/generate_sketch_data.py -------------------------------------------------------------------------------- /scripts/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/scripts/inference.py -------------------------------------------------------------------------------- /scripts/pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/scripts/pretrain.py -------------------------------------------------------------------------------- /scripts/style_mixing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/scripts/style_mixing.py -------------------------------------------------------------------------------- /scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/scripts/train.py -------------------------------------------------------------------------------- /training/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /training/coach.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/training/coach.py -------------------------------------------------------------------------------- /training/ranger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/training/ranger.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/utils/common.py -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/inference_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/utils/inference_utils.py -------------------------------------------------------------------------------- /utils/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/utils/train_utils.py -------------------------------------------------------------------------------- /utils/wandb_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/utils/wandb_utils.py -------------------------------------------------------------------------------- /video_editing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/video_editing.py -------------------------------------------------------------------------------- /webUI/app_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/webUI/app_task.py -------------------------------------------------------------------------------- /webUI/styleganex_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/williamyang1991/StyleGANEX/HEAD/webUI/styleganex_model.py --------------------------------------------------------------------------------