gitextract_xyiv4zse/ ├── .gitignore ├── README.md ├── animate.py ├── augmentation.py ├── config/ │ └── mix-resolution.yml ├── demo.py ├── environment.yaml ├── frames_dataset.py ├── logger.py ├── modules/ │ ├── dense_motion.py │ ├── discriminator.py │ ├── generator.py │ ├── hopenet.py │ ├── keypoint_detector.py │ ├── model.py │ └── util.py ├── run_demo.sh ├── sync_batchnorm/ │ ├── __init__.py │ ├── batchnorm.py │ ├── comm.py │ ├── replicate.py │ └── unittest.py └── upsampler/ ├── 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 ├── 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