gitextract_zvvj76bx/

├── .gitignore
├── LICENSE
├── README.md
├── basicsr/
│   ├── VERSION
│   ├── __init__.py
│   ├── archs/
│   │   ├── __init__.py
│   │   ├── arcface_arch.py
│   │   ├── arch_util.py
│   │   ├── codeformer_arch.py
│   │   ├── rrdbnet_arch.py
│   │   ├── vgg_arch.py
│   │   └── vqgan_arch.py
│   ├── data/
│   │   ├── __init__.py
│   │   ├── data_sampler.py
│   │   ├── data_util.py
│   │   ├── ffhq_blind_dataset.py
│   │   ├── ffhq_blind_joint_dataset.py
│   │   ├── gaussian_kernels.py
│   │   ├── paired_image_dataset.py
│   │   ├── prefetch_dataloader.py
│   │   └── transforms.py
│   ├── losses/
│   │   ├── __init__.py
│   │   ├── loss_util.py
│   │   └── losses.py
│   ├── metrics/
│   │   ├── __init__.py
│   │   ├── metric_util.py
│   │   └── psnr_ssim.py
│   ├── models/
│   │   ├── __init__.py
│   │   ├── base_model.py
│   │   ├── codeformer_idx_model.py
│   │   ├── codeformer_joint_model.py
│   │   ├── codeformer_model.py
│   │   ├── lr_scheduler.py
│   │   ├── sr_model.py
│   │   └── vqgan_model.py
│   ├── ops/
│   │   ├── __init__.py
│   │   ├── dcn/
│   │   │   ├── __init__.py
│   │   │   ├── deform_conv.py
│   │   │   └── src/
│   │   │       ├── deform_conv_cuda.cpp
│   │   │       ├── deform_conv_cuda_kernel.cu
│   │   │       └── deform_conv_ext.cpp
│   │   ├── fused_act/
│   │   │   ├── __init__.py
│   │   │   ├── fused_act.py
│   │   │   └── src/
│   │   │       ├── fused_bias_act.cpp
│   │   │       └── fused_bias_act_kernel.cu
│   │   └── upfirdn2d/
│   │       ├── __init__.py
│   │       ├── src/
│   │       │   ├── upfirdn2d.cpp
│   │       │   └── upfirdn2d_kernel.cu
│   │       └── upfirdn2d.py
│   ├── setup.py
│   ├── train.py
│   └── utils/
│       ├── __init__.py
│       ├── dist_util.py
│       ├── download_util.py
│       ├── file_client.py
│       ├── img_util.py
│       ├── lmdb_util.py
│       ├── logger.py
│       ├── matlab_functions.py
│       ├── misc.py
│       ├── options.py
│       ├── realesrgan_utils.py
│       ├── registry.py
│       └── video_util.py
├── docs/
│   ├── history_changelog.md
│   ├── train.md
│   └── train_CN.md
├── facelib/
│   ├── detection/
│   │   ├── __init__.py
│   │   ├── align_trans.py
│   │   ├── matlab_cp2tform.py
│   │   ├── retinaface/
│   │   │   ├── retinaface.py
│   │   │   ├── retinaface_net.py
│   │   │   └── retinaface_utils.py
│   │   └── yolov5face/
│   │       ├── __init__.py
│   │       ├── face_detector.py
│   │       ├── models/
│   │       │   ├── __init__.py
│   │       │   ├── common.py
│   │       │   ├── experimental.py
│   │       │   ├── yolo.py
│   │       │   ├── yolov5l.yaml
│   │       │   └── yolov5n.yaml
│   │       └── utils/
│   │           ├── __init__.py
│   │           ├── autoanchor.py
│   │           ├── datasets.py
│   │           ├── extract_ckpt.py
│   │           ├── general.py
│   │           └── torch_utils.py
│   ├── parsing/
│   │   ├── __init__.py
│   │   ├── bisenet.py
│   │   ├── parsenet.py
│   │   └── resnet.py
│   └── utils/
│       ├── __init__.py
│       ├── face_restoration_helper.py
│       ├── face_utils.py
│       └── misc.py
├── inference_codeformer.py
├── inference_colorization.py
├── inference_inpainting.py
├── options/
│   ├── CodeFormer_colorization.yml
│   ├── CodeFormer_inpainting.yml
│   ├── CodeFormer_stage2.yml
│   ├── CodeFormer_stage3.yml
│   └── VQGAN_512_ds32_nearest_stage1.yml
├── requirements.txt
├── scripts/
│   ├── crop_align_face.py
│   ├── download_pretrained_models.py
│   ├── download_pretrained_models_from_gdrive.py
│   ├── generate_latent_gt.py
│   └── inference_vqgan.py
├── web-demos/
│   ├── hugging_face/
│   │   └── app.py
│   └── replicate/
│       ├── cog.yaml
│       └── predict.py
└── weights/
    ├── CodeFormer/
    │   └── .gitkeep
    ├── README.md
    └── facelib/
        └── .gitkeep