├── .gitignore ├── LICENSE ├── README.md ├── assets ├── CodeFormer_logo.png ├── color_enhancement_result1.png ├── color_enhancement_result2.png ├── imgsli_1.jpg ├── imgsli_2.jpg ├── imgsli_3.jpg ├── inpainting_result1.png ├── inpainting_result2.png ├── network.jpg ├── restoration_result1.png ├── restoration_result2.png ├── restoration_result3.png └── restoration_result4.png ├── 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 ├── inputs ├── cropped_faces │ ├── 0143.png │ ├── 0240.png │ ├── 0342.png │ ├── 0345.png │ ├── 0368.png │ ├── 0412.png │ ├── 0444.png │ ├── 0478.png │ ├── 0500.png │ ├── 0599.png │ ├── 0717.png │ ├── 0720.png │ ├── 0729.png │ ├── 0763.png │ ├── 0770.png │ ├── 0777.png │ ├── 0885.png │ ├── 0934.png │ ├── Solvay_conference_1927_0018.png │ └── Solvay_conference_1927_2_16.png ├── gray_faces │ ├── 067_David_Beckham_00.png │ ├── 089_Miley_Cyrus_00.png │ ├── 099_Victoria_Beckham_00.png │ ├── 111_Alexa_Chung_00.png │ ├── 132_Robert_Downey_Jr_00.png │ ├── 158_Jimmy_Fallon_00.png │ ├── 161_Zac_Efron_00.png │ ├── 169_John_Lennon_00.png │ ├── 170_Marilyn_Monroe_00.png │ ├── Einstein01.png │ ├── Einstein02.png │ ├── Hepburn01.png │ └── Hepburn02.png ├── masked_faces │ ├── 00105.png │ ├── 00108.png │ ├── 00169.png │ ├── 00588.png │ └── 00664.png └── whole_imgs │ ├── 00.jpg │ ├── 01.jpg │ ├── 02.png │ ├── 03.jpg │ ├── 04.jpg │ ├── 05.jpg │ └── 06.png ├── 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 /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/README.md -------------------------------------------------------------------------------- /assets/CodeFormer_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/assets/CodeFormer_logo.png -------------------------------------------------------------------------------- /assets/color_enhancement_result1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/assets/color_enhancement_result1.png -------------------------------------------------------------------------------- /assets/color_enhancement_result2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/assets/color_enhancement_result2.png -------------------------------------------------------------------------------- /assets/imgsli_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/assets/imgsli_1.jpg -------------------------------------------------------------------------------- /assets/imgsli_2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/assets/imgsli_2.jpg -------------------------------------------------------------------------------- /assets/imgsli_3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/assets/imgsli_3.jpg -------------------------------------------------------------------------------- /assets/inpainting_result1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/assets/inpainting_result1.png -------------------------------------------------------------------------------- /assets/inpainting_result2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/assets/inpainting_result2.png -------------------------------------------------------------------------------- /assets/network.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/assets/network.jpg -------------------------------------------------------------------------------- /assets/restoration_result1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/assets/restoration_result1.png -------------------------------------------------------------------------------- /assets/restoration_result2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/assets/restoration_result2.png -------------------------------------------------------------------------------- /assets/restoration_result3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/assets/restoration_result3.png -------------------------------------------------------------------------------- /assets/restoration_result4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/assets/restoration_result4.png -------------------------------------------------------------------------------- /basicsr/VERSION: -------------------------------------------------------------------------------- 1 | 1.3.2 2 | -------------------------------------------------------------------------------- /basicsr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/__init__.py -------------------------------------------------------------------------------- /basicsr/archs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/archs/__init__.py -------------------------------------------------------------------------------- /basicsr/archs/arcface_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/archs/arcface_arch.py -------------------------------------------------------------------------------- /basicsr/archs/arch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/archs/arch_util.py -------------------------------------------------------------------------------- /basicsr/archs/codeformer_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/archs/codeformer_arch.py -------------------------------------------------------------------------------- /basicsr/archs/rrdbnet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/archs/rrdbnet_arch.py -------------------------------------------------------------------------------- /basicsr/archs/vgg_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/archs/vgg_arch.py -------------------------------------------------------------------------------- /basicsr/archs/vqgan_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/archs/vqgan_arch.py -------------------------------------------------------------------------------- /basicsr/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/data/__init__.py -------------------------------------------------------------------------------- /basicsr/data/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/data/data_sampler.py -------------------------------------------------------------------------------- /basicsr/data/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/data/data_util.py -------------------------------------------------------------------------------- /basicsr/data/ffhq_blind_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/data/ffhq_blind_dataset.py -------------------------------------------------------------------------------- /basicsr/data/ffhq_blind_joint_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/data/ffhq_blind_joint_dataset.py -------------------------------------------------------------------------------- /basicsr/data/gaussian_kernels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/data/gaussian_kernels.py -------------------------------------------------------------------------------- /basicsr/data/paired_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/data/paired_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/prefetch_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/data/prefetch_dataloader.py -------------------------------------------------------------------------------- /basicsr/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/data/transforms.py -------------------------------------------------------------------------------- /basicsr/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/losses/__init__.py -------------------------------------------------------------------------------- /basicsr/losses/loss_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/losses/loss_util.py -------------------------------------------------------------------------------- /basicsr/losses/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/losses/losses.py -------------------------------------------------------------------------------- /basicsr/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/metrics/__init__.py -------------------------------------------------------------------------------- /basicsr/metrics/metric_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/metrics/metric_util.py -------------------------------------------------------------------------------- /basicsr/metrics/psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/metrics/psnr_ssim.py -------------------------------------------------------------------------------- /basicsr/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/models/__init__.py -------------------------------------------------------------------------------- /basicsr/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/models/base_model.py -------------------------------------------------------------------------------- /basicsr/models/codeformer_idx_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/models/codeformer_idx_model.py -------------------------------------------------------------------------------- /basicsr/models/codeformer_joint_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/models/codeformer_joint_model.py -------------------------------------------------------------------------------- /basicsr/models/codeformer_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/models/codeformer_model.py -------------------------------------------------------------------------------- /basicsr/models/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/models/lr_scheduler.py -------------------------------------------------------------------------------- /basicsr/models/sr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/models/sr_model.py -------------------------------------------------------------------------------- /basicsr/models/vqgan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/models/vqgan_model.py -------------------------------------------------------------------------------- /basicsr/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /basicsr/ops/dcn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/ops/dcn/__init__.py -------------------------------------------------------------------------------- /basicsr/ops/dcn/deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/ops/dcn/deform_conv.py -------------------------------------------------------------------------------- /basicsr/ops/dcn/src/deform_conv_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/ops/dcn/src/deform_conv_cuda.cpp -------------------------------------------------------------------------------- /basicsr/ops/dcn/src/deform_conv_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/ops/dcn/src/deform_conv_cuda_kernel.cu -------------------------------------------------------------------------------- /basicsr/ops/dcn/src/deform_conv_ext.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/ops/dcn/src/deform_conv_ext.cpp -------------------------------------------------------------------------------- /basicsr/ops/fused_act/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/ops/fused_act/__init__.py -------------------------------------------------------------------------------- /basicsr/ops/fused_act/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/ops/fused_act/fused_act.py -------------------------------------------------------------------------------- /basicsr/ops/fused_act/src/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/ops/fused_act/src/fused_bias_act.cpp -------------------------------------------------------------------------------- /basicsr/ops/fused_act/src/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/ops/fused_act/src/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/ops/upfirdn2d/__init__.py -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/src/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/ops/upfirdn2d/src/upfirdn2d.cpp -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/src/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/ops/upfirdn2d/src/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/ops/upfirdn2d/upfirdn2d.py -------------------------------------------------------------------------------- /basicsr/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/setup.py -------------------------------------------------------------------------------- /basicsr/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/train.py -------------------------------------------------------------------------------- /basicsr/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/utils/__init__.py -------------------------------------------------------------------------------- /basicsr/utils/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/utils/dist_util.py -------------------------------------------------------------------------------- /basicsr/utils/download_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/utils/download_util.py -------------------------------------------------------------------------------- /basicsr/utils/file_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/utils/file_client.py -------------------------------------------------------------------------------- /basicsr/utils/img_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/utils/img_util.py -------------------------------------------------------------------------------- /basicsr/utils/lmdb_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/utils/lmdb_util.py -------------------------------------------------------------------------------- /basicsr/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/utils/logger.py -------------------------------------------------------------------------------- /basicsr/utils/matlab_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/utils/matlab_functions.py -------------------------------------------------------------------------------- /basicsr/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/utils/misc.py -------------------------------------------------------------------------------- /basicsr/utils/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/utils/options.py -------------------------------------------------------------------------------- /basicsr/utils/realesrgan_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/utils/realesrgan_utils.py -------------------------------------------------------------------------------- /basicsr/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/utils/registry.py -------------------------------------------------------------------------------- /basicsr/utils/video_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/basicsr/utils/video_util.py -------------------------------------------------------------------------------- /docs/history_changelog.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/docs/history_changelog.md -------------------------------------------------------------------------------- /docs/train.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/docs/train.md -------------------------------------------------------------------------------- /docs/train_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/docs/train_CN.md -------------------------------------------------------------------------------- /facelib/detection/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/detection/__init__.py -------------------------------------------------------------------------------- /facelib/detection/align_trans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/detection/align_trans.py -------------------------------------------------------------------------------- /facelib/detection/matlab_cp2tform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/detection/matlab_cp2tform.py -------------------------------------------------------------------------------- /facelib/detection/retinaface/retinaface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/detection/retinaface/retinaface.py -------------------------------------------------------------------------------- /facelib/detection/retinaface/retinaface_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/detection/retinaface/retinaface_net.py -------------------------------------------------------------------------------- /facelib/detection/retinaface/retinaface_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/detection/retinaface/retinaface_utils.py -------------------------------------------------------------------------------- /facelib/detection/yolov5face/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /facelib/detection/yolov5face/face_detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/detection/yolov5face/face_detector.py -------------------------------------------------------------------------------- /facelib/detection/yolov5face/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /facelib/detection/yolov5face/models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/detection/yolov5face/models/common.py -------------------------------------------------------------------------------- /facelib/detection/yolov5face/models/experimental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/detection/yolov5face/models/experimental.py -------------------------------------------------------------------------------- /facelib/detection/yolov5face/models/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/detection/yolov5face/models/yolo.py -------------------------------------------------------------------------------- /facelib/detection/yolov5face/models/yolov5l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/detection/yolov5face/models/yolov5l.yaml -------------------------------------------------------------------------------- /facelib/detection/yolov5face/models/yolov5n.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/detection/yolov5face/models/yolov5n.yaml -------------------------------------------------------------------------------- /facelib/detection/yolov5face/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /facelib/detection/yolov5face/utils/autoanchor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/detection/yolov5face/utils/autoanchor.py -------------------------------------------------------------------------------- /facelib/detection/yolov5face/utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/detection/yolov5face/utils/datasets.py -------------------------------------------------------------------------------- /facelib/detection/yolov5face/utils/extract_ckpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/detection/yolov5face/utils/extract_ckpt.py -------------------------------------------------------------------------------- /facelib/detection/yolov5face/utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/detection/yolov5face/utils/general.py -------------------------------------------------------------------------------- /facelib/detection/yolov5face/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/detection/yolov5face/utils/torch_utils.py -------------------------------------------------------------------------------- /facelib/parsing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/parsing/__init__.py -------------------------------------------------------------------------------- /facelib/parsing/bisenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/parsing/bisenet.py -------------------------------------------------------------------------------- /facelib/parsing/parsenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/parsing/parsenet.py -------------------------------------------------------------------------------- /facelib/parsing/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/parsing/resnet.py -------------------------------------------------------------------------------- /facelib/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/utils/__init__.py -------------------------------------------------------------------------------- /facelib/utils/face_restoration_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/utils/face_restoration_helper.py -------------------------------------------------------------------------------- /facelib/utils/face_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/utils/face_utils.py -------------------------------------------------------------------------------- /facelib/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/facelib/utils/misc.py -------------------------------------------------------------------------------- /inference_codeformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inference_codeformer.py -------------------------------------------------------------------------------- /inference_colorization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inference_colorization.py -------------------------------------------------------------------------------- /inference_inpainting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inference_inpainting.py -------------------------------------------------------------------------------- /inputs/cropped_faces/0143.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/cropped_faces/0143.png -------------------------------------------------------------------------------- /inputs/cropped_faces/0240.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/cropped_faces/0240.png -------------------------------------------------------------------------------- /inputs/cropped_faces/0342.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/cropped_faces/0342.png -------------------------------------------------------------------------------- /inputs/cropped_faces/0345.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/cropped_faces/0345.png -------------------------------------------------------------------------------- /inputs/cropped_faces/0368.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/cropped_faces/0368.png -------------------------------------------------------------------------------- /inputs/cropped_faces/0412.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/cropped_faces/0412.png -------------------------------------------------------------------------------- /inputs/cropped_faces/0444.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/cropped_faces/0444.png -------------------------------------------------------------------------------- /inputs/cropped_faces/0478.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/cropped_faces/0478.png -------------------------------------------------------------------------------- /inputs/cropped_faces/0500.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/cropped_faces/0500.png -------------------------------------------------------------------------------- /inputs/cropped_faces/0599.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/cropped_faces/0599.png -------------------------------------------------------------------------------- /inputs/cropped_faces/0717.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/cropped_faces/0717.png -------------------------------------------------------------------------------- /inputs/cropped_faces/0720.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/cropped_faces/0720.png -------------------------------------------------------------------------------- /inputs/cropped_faces/0729.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/cropped_faces/0729.png -------------------------------------------------------------------------------- /inputs/cropped_faces/0763.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/cropped_faces/0763.png -------------------------------------------------------------------------------- /inputs/cropped_faces/0770.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/cropped_faces/0770.png -------------------------------------------------------------------------------- /inputs/cropped_faces/0777.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/cropped_faces/0777.png -------------------------------------------------------------------------------- /inputs/cropped_faces/0885.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/cropped_faces/0885.png -------------------------------------------------------------------------------- /inputs/cropped_faces/0934.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/cropped_faces/0934.png -------------------------------------------------------------------------------- /inputs/cropped_faces/Solvay_conference_1927_0018.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/cropped_faces/Solvay_conference_1927_0018.png -------------------------------------------------------------------------------- /inputs/cropped_faces/Solvay_conference_1927_2_16.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/cropped_faces/Solvay_conference_1927_2_16.png -------------------------------------------------------------------------------- /inputs/gray_faces/067_David_Beckham_00.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/gray_faces/067_David_Beckham_00.png -------------------------------------------------------------------------------- /inputs/gray_faces/089_Miley_Cyrus_00.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/gray_faces/089_Miley_Cyrus_00.png -------------------------------------------------------------------------------- /inputs/gray_faces/099_Victoria_Beckham_00.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/gray_faces/099_Victoria_Beckham_00.png -------------------------------------------------------------------------------- /inputs/gray_faces/111_Alexa_Chung_00.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/gray_faces/111_Alexa_Chung_00.png -------------------------------------------------------------------------------- /inputs/gray_faces/132_Robert_Downey_Jr_00.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/gray_faces/132_Robert_Downey_Jr_00.png -------------------------------------------------------------------------------- /inputs/gray_faces/158_Jimmy_Fallon_00.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/gray_faces/158_Jimmy_Fallon_00.png -------------------------------------------------------------------------------- /inputs/gray_faces/161_Zac_Efron_00.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/gray_faces/161_Zac_Efron_00.png -------------------------------------------------------------------------------- /inputs/gray_faces/169_John_Lennon_00.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/gray_faces/169_John_Lennon_00.png -------------------------------------------------------------------------------- /inputs/gray_faces/170_Marilyn_Monroe_00.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/gray_faces/170_Marilyn_Monroe_00.png -------------------------------------------------------------------------------- /inputs/gray_faces/Einstein01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/gray_faces/Einstein01.png -------------------------------------------------------------------------------- /inputs/gray_faces/Einstein02.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/gray_faces/Einstein02.png -------------------------------------------------------------------------------- /inputs/gray_faces/Hepburn01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/gray_faces/Hepburn01.png -------------------------------------------------------------------------------- /inputs/gray_faces/Hepburn02.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/gray_faces/Hepburn02.png -------------------------------------------------------------------------------- /inputs/masked_faces/00105.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/masked_faces/00105.png -------------------------------------------------------------------------------- /inputs/masked_faces/00108.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/masked_faces/00108.png -------------------------------------------------------------------------------- /inputs/masked_faces/00169.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/masked_faces/00169.png -------------------------------------------------------------------------------- /inputs/masked_faces/00588.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/masked_faces/00588.png -------------------------------------------------------------------------------- /inputs/masked_faces/00664.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/masked_faces/00664.png -------------------------------------------------------------------------------- /inputs/whole_imgs/00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/whole_imgs/00.jpg -------------------------------------------------------------------------------- /inputs/whole_imgs/01.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/whole_imgs/01.jpg -------------------------------------------------------------------------------- /inputs/whole_imgs/02.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/whole_imgs/02.png -------------------------------------------------------------------------------- /inputs/whole_imgs/03.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/whole_imgs/03.jpg -------------------------------------------------------------------------------- /inputs/whole_imgs/04.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/whole_imgs/04.jpg -------------------------------------------------------------------------------- /inputs/whole_imgs/05.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/whole_imgs/05.jpg -------------------------------------------------------------------------------- /inputs/whole_imgs/06.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/inputs/whole_imgs/06.png -------------------------------------------------------------------------------- /options/CodeFormer_colorization.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/options/CodeFormer_colorization.yml -------------------------------------------------------------------------------- /options/CodeFormer_inpainting.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/options/CodeFormer_inpainting.yml -------------------------------------------------------------------------------- /options/CodeFormer_stage2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/options/CodeFormer_stage2.yml -------------------------------------------------------------------------------- /options/CodeFormer_stage3.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/options/CodeFormer_stage3.yml -------------------------------------------------------------------------------- /options/VQGAN_512_ds32_nearest_stage1.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/options/VQGAN_512_ds32_nearest_stage1.yml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/crop_align_face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/scripts/crop_align_face.py -------------------------------------------------------------------------------- /scripts/download_pretrained_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/scripts/download_pretrained_models.py -------------------------------------------------------------------------------- /scripts/download_pretrained_models_from_gdrive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/scripts/download_pretrained_models_from_gdrive.py -------------------------------------------------------------------------------- /scripts/generate_latent_gt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/scripts/generate_latent_gt.py -------------------------------------------------------------------------------- /scripts/inference_vqgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/scripts/inference_vqgan.py -------------------------------------------------------------------------------- /web-demos/hugging_face/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/web-demos/hugging_face/app.py -------------------------------------------------------------------------------- /web-demos/replicate/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/web-demos/replicate/cog.yaml -------------------------------------------------------------------------------- /web-demos/replicate/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/web-demos/replicate/predict.py -------------------------------------------------------------------------------- /weights/CodeFormer/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /weights/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/CodeFormer/HEAD/weights/README.md -------------------------------------------------------------------------------- /weights/facelib/.gitkeep: -------------------------------------------------------------------------------- 1 | --------------------------------------------------------------------------------