gitextract_ofufjmnn/ ├── .circleci/ │ ├── config.yml │ ├── docker/ │ │ └── Dockerfile │ └── test.yml ├── .dele.yml ├── .dev_scripts/ │ ├── README.md │ ├── create_ceph_configs.py │ ├── doc_link_checker.py │ ├── download_models.py │ ├── inference_benchmark.sh │ ├── job_watcher.py │ ├── metric_mapping.py │ ├── task_mapping.py │ ├── test_benchmark.py │ ├── test_benchmark.yml │ ├── train_benchmark.py │ ├── train_benchmark.yml │ ├── update_config_readme.py │ ├── update_model_index.py │ ├── update_ut.py │ └── utils/ │ ├── __init__.py │ ├── job_util.py │ └── modelindex.py ├── .github/ │ ├── CODE_OF_CONDUCT.md │ ├── CONTRIBUTING.md │ ├── ISSUE_TEMPLATE/ │ │ ├── 1-bug-report.yml │ │ ├── 2-feature-request.yml │ │ ├── 3-new-model.yml │ │ ├── 4-documentation.yml │ │ └── config.yml │ ├── pull_request_template.md │ └── workflows/ │ ├── lint.yml │ ├── merge_stage_test.yml │ ├── pr_stage_test.yml │ ├── publish-to-pypi.yml │ └── test_mim.yml ├── .gitignore ├── .owners.yml ├── .pre-commit-config.yaml ├── .pylintrc ├── .readthedocs.yml ├── CITATION.cff ├── LICENSE ├── MANIFEST.in ├── README.md ├── README_zh-CN.md ├── configs/ │ ├── _base_/ │ │ ├── datasets/ │ │ │ ├── basicvsr_test_config.py │ │ │ ├── celeba.py │ │ │ ├── cifar10_noaug.py │ │ │ ├── cifar10_nopad.py │ │ │ ├── comp1k.py │ │ │ ├── deblurring-defocus_test_config.py │ │ │ ├── deblurring-motion_test_config.py │ │ │ ├── decompression_test_config.py │ │ │ ├── denoising-gaussian_color_test_config.py │ │ │ ├── denoising-gaussian_gray_test_config.py │ │ │ ├── denoising-real_test_config.py │ │ │ ├── deraining_test_config.py │ │ │ ├── ffhq_flip.py │ │ │ ├── grow_scale_imgs_ffhq_styleganv1.py │ │ │ ├── imagenet_128.py │ │ │ ├── imagenet_256.py │ │ │ ├── imagenet_512.py │ │ │ ├── imagenet_64.py │ │ │ ├── imagenet_noaug_128.py │ │ │ ├── liif_test_config.py │ │ │ ├── lsun_stylegan.py │ │ │ ├── paired_imgs_256x256_crop.py │ │ │ ├── places.py │ │ │ ├── sisr_x2_test_config.py │ │ │ ├── sisr_x3_test_config.py │ │ │ ├── sisr_x4_test_config.py │ │ │ ├── tdan_test_config.py │ │ │ ├── unconditional_imgs_128x128.py │ │ │ ├── unconditional_imgs_64x64.py │ │ │ ├── unconditional_imgs_flip_512x512.py │ │ │ ├── unconditional_imgs_flip_lanczos_resize_256x256.py │ │ │ └── unpaired_imgs_256x256.py │ │ ├── default_runtime.py │ │ ├── gen_default_runtime.py │ │ ├── inpaint_default_runtime.py │ │ ├── matting_default_runtime.py │ │ ├── models/ │ │ │ ├── base_cyclegan.py │ │ │ ├── base_deepfillv1.py │ │ │ ├── base_deepfillv2.py │ │ │ ├── base_edvr.py │ │ │ ├── base_gl.py │ │ │ ├── base_glean.py │ │ │ ├── base_liif.py │ │ │ ├── base_pconv.py │ │ │ ├── base_pix2pix.py │ │ │ ├── base_styleganv1.py │ │ │ ├── base_styleganv2.py │ │ │ ├── base_styleganv3.py │ │ │ ├── base_tof.py │ │ │ ├── biggan/ │ │ │ │ └── base_biggan_128x128.py │ │ │ ├── dcgan/ │ │ │ │ ├── base_dcgan_128x128.py │ │ │ │ └── base_dcgan_64x64.py │ │ │ ├── sagan/ │ │ │ │ ├── base_sagan_128x128.py │ │ │ │ └── base_sagan_32x32.py │ │ │ └── sngan_proj/ │ │ │ ├── base_sngan_proj_128x128.py │ │ │ └── base_sngan_proj_32x32.py │ │ └── schedules/ │ │ └── .gitkeep │ ├── animatediff/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── animatediff_Lyriel.py │ │ ├── animatediff_MajicMix.py │ │ ├── animatediff_RcnzCartoon.py │ │ ├── animatediff_RealisticVision.py │ │ ├── animatediff_RealisticVision_v2.py │ │ ├── animatediff_ToonYou.py │ │ └── metafile.yml │ ├── aot_gan/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── aot-gan_smpgan_4xb4_places-512x512.py │ │ └── metafile.yml │ ├── basicvsr/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── basicvsr_2xb4_reds4.py │ │ ├── basicvsr_2xb4_vimeo90k-bd.py │ │ ├── basicvsr_2xb4_vimeo90k-bi.py │ │ └── metafile.yml │ ├── basicvsr_pp/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── basicvsr-pp_c128n25_600k_ntire-decompress-track1.py │ │ ├── basicvsr-pp_c128n25_600k_ntire-decompress-track2.py │ │ ├── basicvsr-pp_c128n25_600k_ntire-decompress-track3.py │ │ ├── basicvsr-pp_c128n25_600k_ntire-vsr.py │ │ ├── basicvsr-pp_c64n7_4xb2-300k_vimeo90k-bd.py │ │ ├── basicvsr-pp_c64n7_4xb2-300k_vimeo90k-bi.py │ │ ├── basicvsr-pp_c64n7_8xb1-600k_reds4.py │ │ └── metafile.yml │ ├── biggan/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── biggan-deep_cvt-hugging-face-rgb_imagenet1k-128x128.py │ │ ├── biggan-deep_cvt-hugging-face_rgb_imagenet1k-256x256.py │ │ ├── biggan-deep_cvt-hugging-face_rgb_imagenet1k-512x512.py │ │ ├── biggan_2xb25-500kiters_cifar10-32x32.py │ │ ├── biggan_ajbrock-sn_8xb32-1500kiters_imagenet1k-128x128.py │ │ ├── biggan_cvt-BigGAN-PyTorch-rgb_imagenet1k-128x128.py │ │ └── metafile.yml │ ├── cain/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── cain_g1b32_1xb5_vimeo90k-triplet.py │ │ └── metafile.yml │ ├── controlnet/ │ │ ├── README.md │ │ ├── controlnet-1xb1-fill50k.py │ │ ├── controlnet-canny.py │ │ ├── controlnet-pose.py │ │ ├── controlnet-seg.py │ │ └── metafile.yml │ ├── controlnet_animation/ │ │ ├── README.md │ │ ├── anythingv3_config.py │ │ └── metafile.yml │ ├── cyclegan/ │ │ ├── README.md │ │ ├── cyclegan_lsgan-id0-resnet-in_1xb1-250kiters_summer2winter.py │ │ ├── cyclegan_lsgan-id0-resnet-in_1xb1-270kiters_horse2zebra.py │ │ ├── cyclegan_lsgan-id0-resnet-in_1xb1-80kiters_facades.py │ │ ├── cyclegan_lsgan-resnet-in_1xb1-250kiters_summer2winter.py │ │ ├── cyclegan_lsgan-resnet-in_1xb1-270kiters_horse2zebra.py │ │ ├── cyclegan_lsgan-resnet-in_1xb1-80kiters_facades.py │ │ └── metafile.yml │ ├── dcgan/ │ │ ├── README.md │ │ ├── dcgan_1xb128-300kiters_celeba-cropped-64.py │ │ ├── dcgan_1xb128-5epoches_lsun-bedroom-64x64.py │ │ ├── dcgan_Glr4e-4_Dlr1e-4_1xb128-5kiters_mnist-64x64.py │ │ └── metafile.yml │ ├── deblurganv2/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── deblurganv2_fpn-inception_1xb1_gopro.py │ │ ├── deblurganv2_fpn-mobilenet_1xb1_gopro.py │ │ └── metafile.yml │ ├── deepfillv1/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── deepfillv1_4xb4_celeba-256x256.py │ │ ├── deepfillv1_8xb2_places-256x256.py │ │ └── metafile.yml │ ├── deepfillv2/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── deepfillv2_8xb2_celeba-256x256.py │ │ ├── deepfillv2_8xb2_places-256x256.py │ │ └── metafile.yml │ ├── dic/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── dic_gan-x8c48b6_4xb2-500k_celeba-hq.py │ │ ├── dic_x8c48b6_4xb2-150k_celeba-hq.py │ │ └── metafile.yml │ ├── diffusers_pipeline/ │ │ ├── README.md │ │ ├── metafile.yml │ │ └── sd_xl_pipeline.py │ ├── dim/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── dim_stage1-v16_1xb1-1000k_comp1k.py │ │ ├── dim_stage1-v16_1xb1-1000k_comp1k_online-merge.py │ │ ├── dim_stage2-v16-pln_1xb1-1000k_comp1k.py │ │ ├── dim_stage3-v16-pln_1xb1-1000k_comp1k.py │ │ └── metafile.yml │ ├── disco_diffusion/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── disco-diffusion_adm-u-finetuned_imagenet-256x256.py │ │ ├── disco-diffusion_adm-u-finetuned_imagenet-512x512.py │ │ ├── disco-diffusion_portrait-generator-v001.py │ │ ├── metafile.yml │ │ └── tutorials.ipynb │ ├── draggan/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── metafile.yml │ │ ├── stylegan2_1024x1024.py │ │ ├── stylegan2_256x256.py │ │ └── stylegan2_512x512.py │ ├── dreambooth/ │ │ ├── README.md │ │ ├── dreambooth-finetune_text_encoder.py │ │ ├── dreambooth-lora-prior_pre.py │ │ ├── dreambooth-lora.py │ │ ├── dreambooth-prior_pre.py │ │ ├── dreambooth.py │ │ └── metafile.yml │ ├── edsr/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── edsr_x2c64b16_1xb16-300k_div2k.py │ │ ├── edsr_x3c64b16_1xb16-300k_div2k.py │ │ ├── edsr_x4c64b16_1xb16-300k_div2k.py │ │ └── metafile.yml │ ├── edvr/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── edvrl_c128b40_8xb8-lr2e-4-600k_reds4.py │ │ ├── edvrl_wotsa-c128b40_8xb8-lr2e-4-600k_reds4.py │ │ ├── edvrm_8xb4-600k_reds.py │ │ ├── edvrm_wotsa_8xb4-600k_reds.py │ │ └── metafile.yml │ ├── eg3d/ │ │ ├── README.md │ │ ├── eg3d_cvt-official-rgb_afhq-512x512.py │ │ ├── eg3d_cvt-official-rgb_ffhq-512x512.py │ │ ├── eg3d_cvt-official-rgb_shapenet-128x128.py │ │ └── metafile.yml │ ├── esrgan/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── esrgan_psnr-x4c64b23g32_1xb16-1000k_div2k.py │ │ ├── esrgan_x4c64b23g32_1xb16-400k_div2k.py │ │ └── metafile.yml │ ├── fastcomposer/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── fastcomposer_8xb16_FFHQ.py │ │ └── metafile.yml │ ├── flavr/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── flavr_in4out1_8xb4_vimeo90k-septuplet.py │ │ └── metafile.yml │ ├── gca/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── baseline_r34_4xb10-200k_comp1k.py │ │ ├── baseline_r34_4xb10-dimaug-200k_comp1k.py │ │ ├── gca_r34_4xb10-200k_comp1k.py │ │ ├── gca_r34_4xb10-dimaug-200k_comp1k.py │ │ └── metafile.yml │ ├── ggan/ │ │ ├── README.md │ │ ├── ggan_dcgan-archi_lr1e-3-1xb128-12Mimgs_celeba-cropped-64x64.py │ │ ├── ggan_dcgan-archi_lr1e-4-1xb64-10Mimgs_celeba-cropped-128x128.py │ │ ├── ggan_lsgan-archi_lr1e-4-1xb128-20Mimgs_lsun-bedroom-64x64.py │ │ └── metafile.yml │ ├── glean/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── glean_in128out1024-fp16_4xb2-300k_ffhq-celeba-hq.py │ │ ├── glean_in128out1024_4xb2-300k_ffhq-celeba-hq.py │ │ ├── glean_x16-fp16_2xb8_ffhq.py │ │ ├── glean_x16_2xb8_cat.py │ │ ├── glean_x16_2xb8_ffhq.py │ │ ├── glean_x8-fp16_2xb8_cat.py │ │ ├── glean_x8_2xb8_cat.py │ │ └── metafile.yml │ ├── global_local/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── gl_8xb12_celeba-256x256.py │ │ ├── gl_8xb12_places-256x256.py │ │ └── metafile.yml │ ├── guided_diffusion/ │ │ ├── README.md │ │ ├── adm-g_ddim25_8xb32_imagenet-256x256.py │ │ ├── adm-g_ddim25_8xb32_imagenet-512x512.py │ │ ├── adm-g_ddim25_8xb32_imagenet-64x64.py │ │ ├── adm_ddim250_8xb32_imagenet-256x256.py │ │ ├── adm_ddim250_8xb32_imagenet-512x512.py │ │ ├── adm_ddim250_8xb32_imagenet-64x64.py │ │ └── metafile.yml │ ├── iconvsr/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── iconvsr_2xb4_reds4.py │ │ ├── iconvsr_2xb4_vimeo90k-bd.py │ │ ├── iconvsr_2xb4_vimeo90k-bi.py │ │ └── metafile.yml │ ├── indexnet/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── indexnet_mobv2-dimaug_1xb16-78k_comp1k.py │ │ ├── indexnet_mobv2_1xb16-78k_comp1k.py │ │ └── metafile.yml │ ├── inst_colorization/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── inst-colorizatioon_full_official_cocostuff-256x256.py │ │ └── metafile.yml │ ├── liif/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── liif-edsr-norm_c64b16_1xb16-1000k_div2k.py │ │ ├── liif-rdn-norm_c64b16_1xb16-1000k_div2k.py │ │ └── metafile.yml │ ├── lsgan/ │ │ ├── README.md │ │ ├── lsgan_dcgan-archi_lr1e-3-1xb128-12Mimgs_celeba-cropped-64x64.py │ │ ├── lsgan_dcgan-archi_lr1e-4-1xb128-12Mimgs_lsun-bedroom-64x64.py │ │ ├── lsgan_dcgan-archi_lr1e-4-1xb64-10Mimgs_celeba-cropped-128x128.py │ │ ├── lsgan_lsgan-archi_lr1e-4-1xb64-10Mimgs_lsun-bedroom-128x128.py │ │ └── metafile.yml │ ├── nafnet/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── metafile.yml │ │ ├── nafnet_c64eb11128mb1db1111_8xb8-lr1e-3-400k_gopro.py │ │ └── nafnet_c64eb2248mb12db2222_8xb8-lr1e-3-400k_sidd.py │ ├── partial_conv/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── metafile.yml │ │ ├── pconv_stage1_8xb12_places-256x256.py │ │ ├── pconv_stage1_8xb1_celeba-256x256.py │ │ ├── pconv_stage2_4xb2_celeba-256x256.py │ │ └── pconv_stage2_4xb2_places-256x256.py │ ├── pggan/ │ │ ├── README.md │ │ ├── metafile.yml │ │ ├── pggan_8xb4-12Mimg_celeba-hq-1024x1024.py │ │ ├── pggan_8xb4-12Mimgs_celeba-cropped-128x128.py │ │ └── pggan_8xb4-12Mimgs_lsun-bedroom-128x128.py │ ├── pix2pix/ │ │ ├── README.md │ │ ├── metafile.yml │ │ ├── pix2pix_vanilla-unet-bn_1xb1-220kiters_aerial2maps.py │ │ ├── pix2pix_vanilla-unet-bn_1xb1-220kiters_maps2aerial.py │ │ ├── pix2pix_vanilla-unet-bn_1xb1-80kiters_facades.py │ │ └── pix2pix_vanilla-unet-bn_wo-jitter-flip-1xb4-190kiters_edges2shoes.py │ ├── positional_encoding_in_gans/ │ │ ├── README.md │ │ ├── metafile.yml │ │ ├── mspie-stylegan2-config-c_c2_8xb3-1100kiters_ffhq-256-512.py │ │ ├── mspie-stylegan2-config-d_c2_8xb3-1100kiters_ffhq-256-512.py │ │ ├── mspie-stylegan2-config-e_c2_8xb3-1100kiters_ffhq-256-512.py │ │ ├── mspie-stylegan2-config-f_c1_8xb2-1600kiters_ffhq-256-1024.py │ │ ├── mspie-stylegan2-config-f_c2_8xb3-1100kiters_ffhq-256-512.py │ │ ├── mspie-stylegan2-config-f_c2_8xb3-1100kiters_ffhq-256-896.py │ │ ├── mspie-stylegan2-config-g_c1_8xb3-1100kiters_ffhq-256-512.py │ │ ├── mspie-stylegan2-config-h_c2_8xb3-1100kiters_ffhq-256-512.py │ │ ├── mspie-stylegan2-config-i_c2_8xb3-1100kiters_ffhq-256-512.py │ │ ├── mspie-stylegan2-config-j_c2_8xb3-1100kiters_ffhq-256-512.py │ │ ├── mspie-stylegan2-config-k_c2_8xb3-1100kiters_ffhq-256-512.py │ │ ├── singan-csg_bohemian.py │ │ ├── singan-csg_fish.py │ │ ├── singan_interp-pad_balloons.py │ │ ├── singan_interp-pad_disc-nobn_balloons.py │ │ ├── singan_interp-pad_disc-nobn_fish.py │ │ ├── singan_spe-dim4_bohemian.py │ │ ├── singan_spe-dim4_fish.py │ │ ├── singan_spe-dim8_bohemian.py │ │ ├── stylegan2_c2_8xb3-1100kiters_ffhq-256x256.py │ │ └── stylegan2_c2_8xb3-1100kiters_ffhq-512x512.py │ ├── rdn/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── metafile.yml │ │ ├── rdn_x2c64b16_1xb16-1000k_div2k.py │ │ ├── rdn_x3c64b16_1xb16-1000k_div2k.py │ │ └── rdn_x4c64b16_1xb16-1000k_div2k.py │ ├── real_basicvsr/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── metafile.yml │ │ ├── realbasicvsr_c64b20-1x30x8_8xb1-lr5e-5-150k_reds.py │ │ └── realbasicvsr_wogan-c64b20-2x30x8_8xb2-lr1e-4-300k_reds.py │ ├── real_esrgan/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── metafile.yml │ │ ├── realesrgan_c64b23g32_4xb12-lr1e-4-400k_df2k-ost.py │ │ └── realesrnet_c64b23g32_4xb12-lr2e-4-1000k_df2k-ost.py │ ├── restormer/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── metafile.yml │ │ ├── restormer_official_dfwb-color-sigma15.py │ │ ├── restormer_official_dfwb-color-sigma25.py │ │ ├── restormer_official_dfwb-color-sigma50.py │ │ ├── restormer_official_dfwb-gray-sigma15.py │ │ ├── restormer_official_dfwb-gray-sigma25.py │ │ ├── restormer_official_dfwb-gray-sigma50.py │ │ ├── restormer_official_dpdd-dual.py │ │ ├── restormer_official_dpdd-single.py │ │ ├── restormer_official_gopro.py │ │ ├── restormer_official_rain13k.py │ │ └── restormer_official_sidd.py │ ├── sagan/ │ │ ├── README.md │ │ ├── metafile.yml │ │ ├── sagan_128_cvt_studioGAN.py │ │ ├── sagan_cvt-studioGAN_cifar10-32x32.py │ │ ├── sagan_wReLUinplace_lr2e-4-ndisc5-1xb64_cifar10-32x32.py │ │ ├── sagan_woReLUinplace-Glr1e-4_Dlr4e-4_noaug-ndisc1-8xb32-bigGAN-sch_imagenet1k-128x128.py │ │ ├── sagan_woReLUinplace_Glr1e-4_Dlr4e-4_ndisc1-4xb64_imagenet1k-128x128.py │ │ └── sagan_woReLUinplace_lr2e-4-ndisc5-1xb64_cifar10-32x32.py │ ├── singan/ │ │ ├── README.md │ │ ├── metafile.yml │ │ ├── singan_balloons.py │ │ ├── singan_bohemian.py │ │ └── singan_fish.py │ ├── sngan_proj/ │ │ ├── README.md │ │ ├── metafile.yml │ │ ├── sngan-proj-cvt-studioGAN_cifar10-32x32.py │ │ ├── sngan-proj-cvt-studioGAN_imagenet1k-128x128.py │ │ ├── sngan-proj_wReLUinplace_Glr2e-4_Dlr5e-5_ndisc5-2xb128_imagenet1k-128x128.py │ │ ├── sngan-proj_wReLUinplace_lr2e-4-ndisc5-1xb64_cifar10-32x32.py │ │ ├── sngan-proj_woReLUinplace_Glr2e-4_Dlr5e-5_ndisc5-2xb128_imagenet1k-128x128.py │ │ └── sngan-proj_woReLUinplace_lr2e-4-ndisc5-1xb64_cifar10-32x32.py │ ├── srcnn/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── metafile.yml │ │ └── srcnn_x4k915_1xb16-1000k_div2k.py │ ├── srgan_resnet/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── metafile.yml │ │ ├── msrresnet_x4c64b16_1xb16-1000k_div2k.py │ │ └── srgan_x4c64b16_1xb16-1000k_div2k.py │ ├── stable_diffusion/ │ │ ├── README.md │ │ ├── metafile.yml │ │ ├── stable-diffusion_ddim_denoisingunet-inpaint.py │ │ ├── stable-diffusion_ddim_denoisingunet-tomesd_5e-1.py │ │ └── stable-diffusion_ddim_denoisingunet.py │ ├── stable_diffusion_xl/ │ │ ├── README.md │ │ ├── metafile.yml │ │ └── stable-diffusion_xl_ddim_denoisingunet.py │ ├── styleganv1/ │ │ ├── README.md │ │ ├── metafile.yml │ │ ├── styleganv1_ffhq-1024x1024_8xb4-25Mimgs.py │ │ └── styleganv1_ffhq-256x256_8xb4-25Mimgs.py │ ├── styleganv2/ │ │ ├── README.md │ │ ├── metafile.yml │ │ ├── stylegan2_c2-PL-R1_8xb4-apex-fp16-no-scaler-800kiters_ffhq-256x256.py │ │ ├── stylegan2_c2-PL-R1_8xb4-fp16-globalG-partialD-no-scaler-800kiters_ffhq-256x256.py │ │ ├── stylegan2_c2-PL_8xb4-fp16-partial-GD-no-scaler-800kiters_ffhq-256x256.py │ │ ├── stylegan2_c2_8xb4-800kiters_ffhq-256x256.py │ │ ├── stylegan2_c2_8xb4-800kiters_lsun-cat-256x256.py │ │ ├── stylegan2_c2_8xb4-800kiters_lsun-church-256x256.py │ │ ├── stylegan2_c2_8xb4-800kiters_lsun-horse-256x256.py │ │ ├── stylegan2_c2_8xb4_ffhq-1024x1024.py │ │ └── stylegan2_c2_8xb4_lsun-car-384x512.py │ ├── styleganv3/ │ │ ├── README.md │ │ ├── metafile.yml │ │ ├── stylegan3-r_ada-gamma3.3_8xb4-fp16_metfaces-1024x1024.py │ │ ├── stylegan3-r_cvt-official-rgb_8xb4_ffhq-1024x1024.py │ │ ├── stylegan3-r_cvt-official-rgb_8xb4_ffhqu-256x256.py │ │ ├── stylegan3-r_cvt-official-rgb_8xb4x8_afhqv2-512x512.py │ │ ├── stylegan3-t_ada-gamma6.6_8xb4-fp16_metfaces-1024x1024.py │ │ ├── stylegan3-t_cvt-official-rgb_8xb4_afhqv2-512x512.py │ │ ├── stylegan3-t_cvt-official-rgb_8xb4_ffhq-1024x1024.py │ │ ├── stylegan3-t_cvt-official-rgb_8xb4_ffhqu-256x256.py │ │ ├── stylegan3-t_gamma2.0_8xb4-fp16-noaug_ffhq-256x256.py │ │ └── stylegan3-t_gamma32.8_8xb4-fp16-noaug_ffhq-1024x1024.py │ ├── swinir/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── metafile.yml │ │ ├── swinir_gan-x2s64w8d6e180_8xb4-lr1e-4-600k_df2k-ost.py │ │ ├── swinir_gan-x4s64w8d6e180_8xb4-lr1e-4-600k_df2k-ost.py │ │ ├── swinir_gan-x4s64w8d9e240_8xb4-lr1e-4-600k_df2k-ost.py │ │ ├── swinir_psnr-x2s64w8d6e180_8xb4-lr1e-4-600k_df2k-ost.py │ │ ├── swinir_psnr-x4s64w8d6e180_8xb4-lr1e-4-600k_df2k-ost.py │ │ ├── swinir_psnr-x4s64w8d9e240_8xb4-lr1e-4-600k_df2k-ost.py │ │ ├── swinir_s126w7d6e180_8xb1-lr2e-4-1600k_dfwb-colorCAR10.py │ │ ├── swinir_s126w7d6e180_8xb1-lr2e-4-1600k_dfwb-colorCAR20.py │ │ ├── swinir_s126w7d6e180_8xb1-lr2e-4-1600k_dfwb-colorCAR30.py │ │ ├── swinir_s126w7d6e180_8xb1-lr2e-4-1600k_dfwb-colorCAR40.py │ │ ├── swinir_s126w7d6e180_8xb1-lr2e-4-1600k_dfwb-grayCAR10.py │ │ ├── swinir_s126w7d6e180_8xb1-lr2e-4-1600k_dfwb-grayCAR20.py │ │ ├── swinir_s126w7d6e180_8xb1-lr2e-4-1600k_dfwb-grayCAR30.py │ │ ├── swinir_s126w7d6e180_8xb1-lr2e-4-1600k_dfwb-grayCAR40.py │ │ ├── swinir_s128w8d6e180_8xb1-lr2e-4-1600k_dfwb-colorDN15.py │ │ ├── swinir_s128w8d6e180_8xb1-lr2e-4-1600k_dfwb-colorDN25.py │ │ ├── swinir_s128w8d6e180_8xb1-lr2e-4-1600k_dfwb-colorDN50.py │ │ ├── swinir_s128w8d6e180_8xb1-lr2e-4-1600k_dfwb-grayDN15.py │ │ ├── swinir_s128w8d6e180_8xb1-lr2e-4-1600k_dfwb-grayDN25.py │ │ ├── swinir_s128w8d6e180_8xb1-lr2e-4-1600k_dfwb-grayDN50.py │ │ ├── swinir_x2s48w8d6e180_8xb4-lr2e-4-500k_div2k.py │ │ ├── swinir_x2s64w8d4e60_8xb4-lr2e-4-500k_div2k.py │ │ ├── swinir_x2s64w8d6e180_8xb4-lr2e-4-500k_df2k.py │ │ ├── swinir_x3s48w8d6e180_8xb4-lr2e-4-500k_div2k.py │ │ ├── swinir_x3s64w8d4e60_8xb4-lr2e-4-500k_div2k.py │ │ ├── swinir_x3s64w8d6e180_8xb4-lr2e-4-500k_df2k.py │ │ ├── swinir_x4s48w8d6e180_8xb4-lr2e-4-500k_div2k.py │ │ ├── swinir_x4s64w8d4e60_8xb4-lr2e-4-500k_div2k.py │ │ └── swinir_x4s64w8d6e180_8xb4-lr2e-4-500k_df2k.py │ ├── tdan/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── metafile.yml │ │ ├── tdan_x4_8xb16-lr1e-4-400k_vimeo90k-bd.py │ │ ├── tdan_x4_8xb16-lr1e-4-400k_vimeo90k-bi.py │ │ ├── tdan_x4ft_8xb16-lr5e-5-400k_vimeo90k-bi.py │ │ └── tdan_x4ft_8xb16-lr5e-5-800k_vimeo90k-bd.py │ ├── textual_inversion/ │ │ ├── README.md │ │ ├── metafile.yml │ │ └── textual_inversion.py │ ├── tof/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── metafile.yml │ │ ├── tof_spynet-chair-wobn_1xb1_vimeo90k-triplet.py │ │ ├── tof_spynet-kitti-wobn_1xb1_vimeo90k-triplet.py │ │ ├── tof_spynet-pytoflow-wobn_1xb1_vimeo90k-triplet.py │ │ ├── tof_spynet-sintel-wobn-clean_1xb1_vimeo90k-triplet.py │ │ ├── tof_spynet-sintel-wobn-final_1xb1_vimeo90k-triplet.py │ │ └── tof_x4_official_vimeo90k.py │ ├── ttsr/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── metafile.yml │ │ ├── ttsr-gan_x4c64b16_1xb9-500k_CUFED.py │ │ └── ttsr-rec_x4c64b16_1xb9-200k_CUFED.py │ ├── vico/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── metafile.yml │ │ └── vico.py │ └── wgan-gp/ │ ├── README.md │ ├── metafile.yml │ ├── wgangp_GN-GP-50_1xb64-160kiters_lsun-bedroom-128x128.py │ └── wgangp_GN_1xb64-160kiters_celeba-cropped-128x128.py ├── demo/ │ ├── README.md │ ├── download_inference_resources.py │ ├── gradio_animatediff.py │ ├── gradio_controlnet_animation.py │ ├── gradio_draggan.py │ ├── gradio_fastcomposer.py │ ├── gradio_inpainting.py │ ├── gradio_vico.py │ ├── mmagic_inference_demo.py │ ├── mmagic_inference_tutorial.ipynb │ ├── singan_demo.py │ └── utils/ │ ├── gradio_utils.py │ └── renderer.py ├── docker/ │ ├── Dockerfile │ └── README.md ├── docs/ │ ├── en/ │ │ ├── .dev_scripts/ │ │ │ ├── update_dataset_zoo.py │ │ │ └── update_model_zoo.py │ │ ├── .gitignore │ │ ├── Makefile │ │ ├── _static/ │ │ │ └── css/ │ │ │ └── readthedocs.css │ │ ├── _templates/ │ │ │ ├── 404.html │ │ │ └── python/ │ │ │ ├── attribute.rst │ │ │ ├── class.rst │ │ │ ├── data.rst │ │ │ ├── exception.rst │ │ │ ├── function.rst │ │ │ ├── method.rst │ │ │ ├── module.rst │ │ │ ├── package.rst │ │ │ └── property.rst │ │ ├── advanced_guides/ │ │ │ ├── data_flow.md │ │ │ ├── data_preprocessor.md │ │ │ ├── evaluator.md │ │ │ └── structures.md │ │ ├── changelog.md │ │ ├── community/ │ │ │ ├── contributing.md │ │ │ └── projects.md │ │ ├── conf.py │ │ ├── device/ │ │ │ └── npu.md │ │ ├── docutils.conf │ │ ├── faq.md │ │ ├── get_started/ │ │ │ ├── install.md │ │ │ ├── overview.md │ │ │ └── quick_run.md │ │ ├── howto/ │ │ │ ├── dataset.md │ │ │ ├── losses.md │ │ │ ├── models.md │ │ │ └── transforms.md │ │ ├── index.rst │ │ ├── make.bat │ │ ├── migration/ │ │ │ ├── amp.md │ │ │ ├── data.md │ │ │ ├── distributed_train.md │ │ │ ├── eval_test.md │ │ │ ├── models.md │ │ │ ├── optimizers.md │ │ │ ├── overview.md │ │ │ ├── runtime.md │ │ │ ├── schedule.md │ │ │ └── visualization.md │ │ ├── switch_language.md │ │ └── user_guides/ │ │ ├── config.md │ │ ├── dataset_prepare.md │ │ ├── deploy.md │ │ ├── inference.md │ │ ├── metrics.md │ │ ├── train_test.md │ │ ├── useful_tools.md │ │ └── visualization.md │ └── zh_cn/ │ ├── .dev_scripts/ │ │ ├── update_dataset_zoo.py │ │ └── update_model_zoo.py │ ├── .gitignore │ ├── Makefile │ ├── _static/ │ │ └── css/ │ │ └── readthedocs.css │ ├── _templates/ │ │ └── 404.html │ ├── advanced_guides/ │ │ ├── data_flow.md │ │ ├── data_preprocessor.md │ │ ├── evaluator.md │ │ └── structures.md │ ├── changelog.md │ ├── community/ │ │ ├── contributing.md │ │ └── projects.md │ ├── conf.py │ ├── device/ │ │ └── npu_zh.md │ ├── faq.md │ ├── get_started/ │ │ ├── install.md │ │ ├── overview.md │ │ └── quick_run.md │ ├── howto/ │ │ ├── dataset.md │ │ ├── losses.md │ │ ├── models.md │ │ └── transforms.md │ ├── index.rst │ ├── make.bat │ ├── migration/ │ │ ├── amp.md │ │ ├── data.md │ │ ├── distributed_train.md │ │ ├── eval_test.md │ │ ├── models.md │ │ ├── optimizers.md │ │ ├── overview.md │ │ ├── runtime.md │ │ ├── schedule.md │ │ └── visualization.md │ ├── stat.py │ ├── switch_language.md │ └── user_guides/ │ ├── config.md │ ├── dataset_prepare.md │ ├── deploy.md │ ├── index.rst │ ├── inference.md │ ├── metrics.md │ ├── train_test.md │ ├── useful_tools.md │ └── visualization.md ├── mmagic/ │ ├── __init__.py │ ├── apis/ │ │ ├── __init__.py │ │ ├── inferencers/ │ │ │ ├── __init__.py │ │ │ ├── base_mmagic_inferencer.py │ │ │ ├── colorization_inferencer.py │ │ │ ├── conditional_inferencer.py │ │ │ ├── controlnet_animation_inferencer.py │ │ │ ├── diffusers_pipeline_inferencer.py │ │ │ ├── eg3d_inferencer.py │ │ │ ├── image_super_resolution_inferencer.py │ │ │ ├── inference_functions.py │ │ │ ├── inpainting_inferencer.py │ │ │ ├── matting_inferencer.py │ │ │ ├── text2image_inferencer.py │ │ │ ├── translation_inferencer.py │ │ │ ├── unconditional_inferencer.py │ │ │ ├── video_interpolation_inferencer.py │ │ │ └── video_restoration_inferencer.py │ │ └── mmagic_inferencer.py │ ├── configs/ │ │ ├── _base_/ │ │ │ ├── datasets/ │ │ │ │ ├── basicvsr_test_config.py │ │ │ │ ├── celeba.py │ │ │ │ ├── cifar10_noaug.py │ │ │ │ ├── cifar10_nopad.py │ │ │ │ ├── comp1k.py │ │ │ │ ├── deblurring-defocus_test_config.py │ │ │ │ ├── deblurring-motion_test_config.py │ │ │ │ ├── decompression_test_config.py │ │ │ │ ├── denoising-gaussian_color_test_config.py │ │ │ │ ├── denoising-gaussian_gray_test_config.py │ │ │ │ ├── denoising-real_test_config.py │ │ │ │ ├── deraining_test_config.py │ │ │ │ ├── ffhq_flip.py │ │ │ │ ├── grow_scale_imgs_ffhq_styleganv1.py │ │ │ │ ├── imagenet_128.py │ │ │ │ ├── imagenet_256.py │ │ │ │ ├── imagenet_512.py │ │ │ │ ├── imagenet_64.py │ │ │ │ ├── imagenet_noaug_128.py │ │ │ │ ├── liif_test_config.py │ │ │ │ ├── lsun_stylegan.py │ │ │ │ ├── paired_imgs_256x256_crop.py │ │ │ │ ├── places.py │ │ │ │ ├── sisr_x2_test_config.py │ │ │ │ ├── sisr_x3_test_config.py │ │ │ │ ├── sisr_x4_test_config.py │ │ │ │ ├── tdan_test_config.py │ │ │ │ ├── unconditional_imgs_128x128.py │ │ │ │ ├── unconditional_imgs_64x64.py │ │ │ │ ├── unconditional_imgs_flip_512x512.py │ │ │ │ ├── unconditional_imgs_flip_lanczos_resize_256x256.py │ │ │ │ └── unpaired_imgs_256x256.py │ │ │ ├── default_runtime.py │ │ │ ├── gen_default_runtime.py │ │ │ ├── inpaint_default_runtime.py │ │ │ ├── matting_default_runtime.py │ │ │ ├── models/ │ │ │ │ ├── base_cyclegan.py │ │ │ │ ├── base_deepfillv1.py │ │ │ │ ├── base_deepfillv2.py │ │ │ │ ├── base_edvr.py │ │ │ │ ├── base_gl.py │ │ │ │ ├── base_glean.py │ │ │ │ ├── base_liif.py │ │ │ │ ├── base_pconv.py │ │ │ │ ├── base_pix2pix.py │ │ │ │ ├── base_styleganv1.py │ │ │ │ ├── base_styleganv2.py │ │ │ │ ├── base_styleganv3.py │ │ │ │ ├── base_tof.py │ │ │ │ ├── biggan/ │ │ │ │ │ └── base_biggan_128x128.py │ │ │ │ ├── dcgan/ │ │ │ │ │ ├── base_dcgan_128x128.py │ │ │ │ │ └── base_dcgan_64x64.py │ │ │ │ ├── sagan/ │ │ │ │ │ ├── base_sagan_128x128.py │ │ │ │ │ └── base_sagan_32x32.py │ │ │ │ └── sngan_proj/ │ │ │ │ ├── base_sngan_proj_128x128.py │ │ │ │ └── base_sngan_proj_32x32.py │ │ │ └── schedules/ │ │ │ └── .gitkeep │ │ ├── biggan/ │ │ │ ├── biggan-deep_cvt-hugging-face-rgb_imagenet1k-128x128.py │ │ │ ├── biggan-deep_cvt-hugging-face_rgb_imagenet1k-256x256.py │ │ │ ├── biggan-deep_cvt-hugging-face_rgb_imagenet1k-512x512.py │ │ │ ├── biggan_2xb25-500kiters_cifar10-32x32.py │ │ │ ├── biggan_ajbrock-sn_8xb32-1500kiters_imagenet1k-128x128.py │ │ │ └── biggan_cvt-BigGAN-PyTorch-rgb_imagenet1k-128x128.py │ │ ├── dreambooth/ │ │ │ ├── dreambooth-finetune_text_encoder.py │ │ │ ├── dreambooth-prior_pre.py │ │ │ ├── dreambooth.py │ │ │ ├── dreambooth_lora-prior_pre.py │ │ │ └── dreambooth_lora.py │ │ ├── eg3d/ │ │ │ ├── eg3d_cvt-official-rgb_afhq-512x512.py │ │ │ ├── eg3d_cvt-official-rgb_ffhq-512x512.py │ │ │ └── eg3d_cvt-official-rgb_shapenet-128x128.py │ │ ├── guided_diffusion/ │ │ │ ├── adm-g_ddim25_8xb32_imagenet_256x256.py │ │ │ ├── adm-g_ddim25_8xb32_imagenet_512x512.py │ │ │ ├── adm-g_ddim25_8xb32_imagenet_64x64.py │ │ │ ├── adm_ddim250_8xb32_imagenet_256x256.py │ │ │ ├── adm_ddim250_8xb32_imagenet_512x512.py │ │ │ └── adm_ddim250_8xb32_imagenet_64x64.py │ │ ├── inst_colorization/ │ │ │ └── inst-colorizatioon_full_official_cocostuff-256x256.py │ │ ├── real_basicvsr/ │ │ │ ├── realbasicvsr_c64b20_1x30x8_8xb1_lr5e_5_150k_reds.py │ │ │ └── realbasicvsr_wogan_c64b20_2x30x8_8xb2_lr1e_4_300k_reds.py │ │ ├── styleganv2/ │ │ │ ├── stylegan2_c2_8xb4_800kiters_ffhq_256x256.py │ │ │ ├── stylegan2_c2_8xb4_800kiters_lsun_cat_256x256.py │ │ │ ├── stylegan2_c2_8xb4_800kiters_lsun_church_256x256.py │ │ │ ├── stylegan2_c2_8xb4_800kiters_lsun_horse_256x256.py │ │ │ ├── stylegan2_c2_8xb4_ffhq_1024x1024.py │ │ │ ├── stylegan2_c2_8xb4_lsun_car_384x512.py │ │ │ ├── stylegan2_c2_PL_8xb4_fp16_partial_GD_no_scaler_800kiters_ffhq_256x256.py │ │ │ ├── stylegan2_c2_PL_R1_8xb4_apex_fp16_no_scaler_800kiters_ffhq_256x256.py │ │ │ └── stylegan2_c2_PL_R1_8xb4_fp16_globalG_partialD_no_scaler_800kiters_ffhq_256x256.py │ │ └── styleganv3/ │ │ ├── stylegan3_r_ada_gamma33_8xb4_fp16_metfaces_1024x1024.py │ │ ├── stylegan3_r_cvt_official_rgb_8xb4_ffhq_1024x1024.py │ │ ├── stylegan3_r_cvt_official_rgb_8xb4_ffhqu_256x256.py │ │ ├── stylegan3_r_cvt_official_rgb_8xb4x8_afhqv2_512x512.py │ │ ├── stylegan3_t_ada_gamma66_8xb4_fp16_metfaces_1024x1024.py │ │ ├── stylegan3_t_cvt_official_rgb_8xb4_afhqv2_512x512.py │ │ ├── stylegan3_t_cvt_official_rgb_8xb4_ffhq_1024x1024.py │ │ ├── stylegan3_t_cvt_official_rgb_8xb4_ffhqu_256x256.py │ │ ├── stylegan3_t_gamma20_8xb4_fp16_noaug_ffhq_256x256.py │ │ └── stylegan3_t_gamma328_8xb4_fp16_noaug_ffhq_1024x1024.py │ ├── datasets/ │ │ ├── __init__.py │ │ ├── basic_conditional_dataset.py │ │ ├── basic_frames_dataset.py │ │ ├── basic_image_dataset.py │ │ ├── categories.py │ │ ├── cifar10_dataset.py │ │ ├── comp1k_dataset.py │ │ ├── controlnet_dataset.py │ │ ├── data_utils.py │ │ ├── dreambooth_dataset.py │ │ ├── grow_scale_image_dataset.py │ │ ├── imagenet_dataset.py │ │ ├── mscoco_dataset.py │ │ ├── paired_image_dataset.py │ │ ├── singan_dataset.py │ │ ├── textual_inversion_dataset.py │ │ ├── transforms/ │ │ │ ├── __init__.py │ │ │ ├── albu_function.py │ │ │ ├── albumentations.py │ │ │ ├── alpha.py │ │ │ ├── aug_frames.py │ │ │ ├── aug_pixel.py │ │ │ ├── aug_shape.py │ │ │ ├── blur_kernels.py │ │ │ ├── crop.py │ │ │ ├── fgbg.py │ │ │ ├── formatting.py │ │ │ ├── generate_assistant.py │ │ │ ├── generate_frame_indices.py │ │ │ ├── get_masked_image.py │ │ │ ├── loading.py │ │ │ ├── matlab_like_resize.py │ │ │ ├── normalization.py │ │ │ ├── random_degradations.py │ │ │ ├── random_down_sampling.py │ │ │ ├── trimap.py │ │ │ └── values.py │ │ └── unpaired_image_dataset.py │ ├── engine/ │ │ ├── __init__.py │ │ ├── hooks/ │ │ │ ├── __init__.py │ │ │ ├── ema.py │ │ │ ├── iter_time_hook.py │ │ │ ├── pggan_fetch_data_hook.py │ │ │ ├── pickle_data_hook.py │ │ │ ├── reduce_lr_scheduler_hook.py │ │ │ └── visualization_hook.py │ │ ├── optimizers/ │ │ │ ├── __init__.py │ │ │ ├── multi_optimizer_constructor.py │ │ │ ├── pggan_optimizer_constructor.py │ │ │ └── singan_optimizer_constructor.py │ │ ├── runner/ │ │ │ ├── __init__.py │ │ │ ├── log_processor.py │ │ │ ├── loop_utils.py │ │ │ └── multi_loops.py │ │ └── schedulers/ │ │ ├── __init__.py │ │ ├── linear_lr_scheduler_with_interval.py │ │ └── reduce_lr_scheduler.py │ ├── evaluation/ │ │ ├── __init__.py │ │ ├── evaluator.py │ │ ├── functional/ │ │ │ ├── __init__.py │ │ │ ├── fid_inception.py │ │ │ ├── gaussian_funcs.py │ │ │ └── inception_utils.py │ │ └── metrics/ │ │ ├── __init__.py │ │ ├── base_gen_metric.py │ │ ├── base_sample_wise_metric.py │ │ ├── connectivity_error.py │ │ ├── equivariance.py │ │ ├── fid.py │ │ ├── gradient_error.py │ │ ├── inception_score.py │ │ ├── mae.py │ │ ├── matting_mse.py │ │ ├── metrics_utils.py │ │ ├── ms_ssim.py │ │ ├── mse.py │ │ ├── niqe.py │ │ ├── niqe_pris_params.npz │ │ ├── ppl.py │ │ ├── precision_and_recall.py │ │ ├── psnr.py │ │ ├── sad.py │ │ ├── snr.py │ │ ├── ssim.py │ │ └── swd.py │ ├── models/ │ │ ├── __init__.py │ │ ├── archs/ │ │ │ ├── __init__.py │ │ │ ├── all_gather_layer.py │ │ │ ├── aspp.py │ │ │ ├── attention_injection.py │ │ │ ├── conv.py │ │ │ ├── downsample.py │ │ │ ├── ensemble.py │ │ │ ├── gated_conv_module.py │ │ │ ├── img_normalize.py │ │ │ ├── linear_module.py │ │ │ ├── lora.py │ │ │ ├── multi_layer_disc.py │ │ │ ├── patch_disc.py │ │ │ ├── resnet.py │ │ │ ├── separable_conv_module.py │ │ │ ├── simple_encoder_decoder.py │ │ │ ├── smpatch_disc.py │ │ │ ├── sr_backbone.py │ │ │ ├── tokenizer.py │ │ │ ├── upsample.py │ │ │ ├── vgg.py │ │ │ └── wrapper.py │ │ ├── base_models/ │ │ │ ├── __init__.py │ │ │ ├── average_model.py │ │ │ ├── base_conditional_gan.py │ │ │ ├── base_edit_model.py │ │ │ ├── base_gan.py │ │ │ ├── base_mattor.py │ │ │ ├── base_translation_model.py │ │ │ ├── basic_interpolator.py │ │ │ ├── one_stage.py │ │ │ └── two_stage.py │ │ ├── data_preprocessors/ │ │ │ ├── __init__.py │ │ │ ├── data_preprocessor.py │ │ │ └── mattor_preprocessor.py │ │ ├── diffusion_schedulers/ │ │ │ ├── __init__.py │ │ │ ├── ddim_scheduler.py │ │ │ └── ddpm_scheduler.py │ │ ├── editors/ │ │ │ ├── __init__.py │ │ │ ├── animatediff/ │ │ │ │ ├── __init__.py │ │ │ │ ├── animatediff.py │ │ │ │ ├── animatediff_utils.py │ │ │ │ ├── attention_3d.py │ │ │ │ ├── motion_module.py │ │ │ │ ├── resnet_3d.py │ │ │ │ ├── unet_3d.py │ │ │ │ └── unet_block.py │ │ │ ├── aotgan/ │ │ │ │ ├── __init__.py │ │ │ │ ├── aot_decoder.py │ │ │ │ ├── aot_encoder.py │ │ │ │ ├── aot_encoder_decoder.py │ │ │ │ ├── aot_inpaintor.py │ │ │ │ └── aot_neck.py │ │ │ ├── arcface/ │ │ │ │ ├── __init__.py │ │ │ │ ├── arcface_modules.py │ │ │ │ ├── id_loss.py │ │ │ │ └── model_irse.py │ │ │ ├── basicvsr/ │ │ │ │ ├── __init__.py │ │ │ │ ├── basicvsr.py │ │ │ │ └── basicvsr_net.py │ │ │ ├── basicvsr_plusplus_net/ │ │ │ │ ├── __init__.py │ │ │ │ └── basicvsr_plusplus_net.py │ │ │ ├── biggan/ │ │ │ │ ├── __init__.py │ │ │ │ ├── biggan.py │ │ │ │ ├── biggan_deep_discriminator.py │ │ │ │ ├── biggan_deep_generator.py │ │ │ │ ├── biggan_discriminator.py │ │ │ │ ├── biggan_generator.py │ │ │ │ ├── biggan_modules.py │ │ │ │ └── biggan_snmodule.py │ │ │ ├── cain/ │ │ │ │ ├── __init__.py │ │ │ │ ├── cain.py │ │ │ │ └── cain_net.py │ │ │ ├── controlnet/ │ │ │ │ ├── __init__.py │ │ │ │ ├── controlnet.py │ │ │ │ └── controlnet_utils.py │ │ │ ├── cyclegan/ │ │ │ │ ├── __init__.py │ │ │ │ ├── cyclegan.py │ │ │ │ ├── cyclegan_generator.py │ │ │ │ └── cyclegan_modules.py │ │ │ ├── dcgan/ │ │ │ │ ├── __init__.py │ │ │ │ ├── dcgan.py │ │ │ │ ├── dcgan_discriminator.py │ │ │ │ └── dcgan_generator.py │ │ │ ├── ddpm/ │ │ │ │ ├── __init__.py │ │ │ │ ├── attention.py │ │ │ │ ├── denoising_unet.py │ │ │ │ ├── embeddings.py │ │ │ │ ├── res_blocks.py │ │ │ │ └── unet_blocks.py │ │ │ ├── deblurganv2/ │ │ │ │ ├── __init__.py │ │ │ │ ├── deblurganv2.py │ │ │ │ ├── deblurganv2_discriminator.py │ │ │ │ ├── deblurganv2_generator.py │ │ │ │ └── deblurganv2_util.py │ │ │ ├── deepfillv1/ │ │ │ │ ├── __init__.py │ │ │ │ ├── contextual_attention.py │ │ │ │ ├── contextual_attention_neck.py │ │ │ │ ├── deepfill_decoder.py │ │ │ │ ├── deepfill_disc.py │ │ │ │ ├── deepfill_encoder.py │ │ │ │ ├── deepfill_refiner.py │ │ │ │ └── deepfillv1.py │ │ │ ├── deepfillv2/ │ │ │ │ ├── __init__.py │ │ │ │ └── two_stage_encoder_decoder.py │ │ │ ├── dic/ │ │ │ │ ├── __init__.py │ │ │ │ ├── dic.py │ │ │ │ ├── dic_net.py │ │ │ │ ├── feedback_hour_glass.py │ │ │ │ └── light_cnn.py │ │ │ ├── dim/ │ │ │ │ ├── __init__.py │ │ │ │ └── dim.py │ │ │ ├── disco_diffusion/ │ │ │ │ ├── __init__.py │ │ │ │ ├── clip_wrapper.py │ │ │ │ ├── disco.py │ │ │ │ ├── guider.py │ │ │ │ └── secondary_model.py │ │ │ ├── dreambooth/ │ │ │ │ ├── __init__.py │ │ │ │ └── dreambooth.py │ │ │ ├── duf/ │ │ │ │ ├── __init__.py │ │ │ │ └── duf.py │ │ │ ├── edsr/ │ │ │ │ ├── __init__.py │ │ │ │ └── edsr_net.py │ │ │ ├── edvr/ │ │ │ │ ├── __init__.py │ │ │ │ ├── edvr.py │ │ │ │ └── edvr_net.py │ │ │ ├── eg3d/ │ │ │ │ ├── __init__.py │ │ │ │ ├── camera.py │ │ │ │ ├── dual_discriminator.py │ │ │ │ ├── eg3d.py │ │ │ │ ├── eg3d_generator.py │ │ │ │ ├── eg3d_modules.py │ │ │ │ ├── eg3d_utils.py │ │ │ │ ├── ray_sampler.py │ │ │ │ └── renderer.py │ │ │ ├── esrgan/ │ │ │ │ ├── __init__.py │ │ │ │ ├── esrgan.py │ │ │ │ └── rrdb_net.py │ │ │ ├── fastcomposer/ │ │ │ │ ├── __init__.py │ │ │ │ ├── fastcomposer.py │ │ │ │ └── fastcomposer_util.py │ │ │ ├── fba/ │ │ │ │ ├── __init__.py │ │ │ │ ├── fba_decoder.py │ │ │ │ └── fba_encoder.py │ │ │ ├── flavr/ │ │ │ │ ├── __init__.py │ │ │ │ ├── flavr.py │ │ │ │ └── flavr_net.py │ │ │ ├── gca/ │ │ │ │ ├── __init__.py │ │ │ │ ├── gca.py │ │ │ │ ├── gca_module.py │ │ │ │ ├── resgca_dec.py │ │ │ │ └── resgca_enc.py │ │ │ ├── ggan/ │ │ │ │ ├── __init__.py │ │ │ │ └── ggan.py │ │ │ ├── glean/ │ │ │ │ ├── __init__.py │ │ │ │ └── glean_styleganv2.py │ │ │ ├── global_local/ │ │ │ │ ├── __init__.py │ │ │ │ ├── gl_decoder.py │ │ │ │ ├── gl_dilation.py │ │ │ │ ├── gl_disc.py │ │ │ │ ├── gl_encoder.py │ │ │ │ ├── gl_encoder_decoder.py │ │ │ │ └── gl_inpaintor.py │ │ │ ├── guided_diffusion/ │ │ │ │ ├── __init__.py │ │ │ │ ├── adm.py │ │ │ │ └── classifier.py │ │ │ ├── iconvsr/ │ │ │ │ ├── __init__.py │ │ │ │ └── iconvsr_net.py │ │ │ ├── indexnet/ │ │ │ │ ├── __init__.py │ │ │ │ ├── indexnet.py │ │ │ │ ├── indexnet_decoder.py │ │ │ │ └── indexnet_encoder.py │ │ │ ├── inst_colorization/ │ │ │ │ ├── __init__.py │ │ │ │ ├── color_utils.py │ │ │ │ ├── colorization_net.py │ │ │ │ ├── fusion_net.py │ │ │ │ ├── inst_colorization.py │ │ │ │ └── weight_layer.py │ │ │ ├── liif/ │ │ │ │ ├── __init__.py │ │ │ │ ├── liif.py │ │ │ │ ├── liif_net.py │ │ │ │ └── mlp_refiner.py │ │ │ ├── lsgan/ │ │ │ │ ├── __init__.py │ │ │ │ ├── lsgan.py │ │ │ │ ├── lsgan_discriminator.py │ │ │ │ └── lsgan_generator.py │ │ │ ├── mspie/ │ │ │ │ ├── __init__.py │ │ │ │ ├── mspie_stylegan2.py │ │ │ │ ├── mspie_stylegan2_discriminator.py │ │ │ │ ├── mspie_stylegan2_generator.py │ │ │ │ ├── mspie_stylegan2_modules.py │ │ │ │ ├── pe_singan.py │ │ │ │ ├── pe_singan_generator.py │ │ │ │ └── positional_encoding.py │ │ │ ├── nafnet/ │ │ │ │ ├── __init__.py │ │ │ │ ├── naf_avgpool2d.py │ │ │ │ ├── naf_layerNorm2d.py │ │ │ │ ├── nafbaseline_net.py │ │ │ │ └── nafnet_net.py │ │ │ ├── pconv/ │ │ │ │ ├── __init__.py │ │ │ │ ├── mask_conv_module.py │ │ │ │ ├── partial_conv.py │ │ │ │ ├── pconv_decoder.py │ │ │ │ ├── pconv_encoder.py │ │ │ │ ├── pconv_encoder_decoder.py │ │ │ │ └── pconv_inpaintor.py │ │ │ ├── pggan/ │ │ │ │ ├── __init__.py │ │ │ │ ├── pggan.py │ │ │ │ ├── pggan_discriminator.py │ │ │ │ ├── pggan_generator.py │ │ │ │ └── pggan_modules.py │ │ │ ├── pix2pix/ │ │ │ │ ├── __init__.py │ │ │ │ ├── pix2pix.py │ │ │ │ ├── pix2pix_generator.py │ │ │ │ └── pix2pix_modules.py │ │ │ ├── plain/ │ │ │ │ ├── __init__.py │ │ │ │ ├── plain_decoder.py │ │ │ │ └── plain_refiner.py │ │ │ ├── rdn/ │ │ │ │ ├── __init__.py │ │ │ │ └── rdn_net.py │ │ │ ├── real_basicvsr/ │ │ │ │ ├── __init__.py │ │ │ │ ├── real_basicvsr.py │ │ │ │ └── real_basicvsr_net.py │ │ │ ├── real_esrgan/ │ │ │ │ ├── __init__.py │ │ │ │ ├── real_esrgan.py │ │ │ │ └── unet_disc.py │ │ │ ├── restormer/ │ │ │ │ ├── __init__.py │ │ │ │ └── restormer_net.py │ │ │ ├── sagan/ │ │ │ │ ├── __init__.py │ │ │ │ ├── sagan.py │ │ │ │ ├── sagan_discriminator.py │ │ │ │ ├── sagan_generator.py │ │ │ │ └── sagan_modules.py │ │ │ ├── singan/ │ │ │ │ ├── __init__.py │ │ │ │ ├── singan.py │ │ │ │ ├── singan_discriminator.py │ │ │ │ ├── singan_generator.py │ │ │ │ └── singan_modules.py │ │ │ ├── srcnn/ │ │ │ │ ├── __init__.py │ │ │ │ └── srcnn_net.py │ │ │ ├── srgan/ │ │ │ │ ├── __init__.py │ │ │ │ ├── modified_vgg.py │ │ │ │ ├── sr_resnet.py │ │ │ │ └── srgan.py │ │ │ ├── stable_diffusion/ │ │ │ │ ├── __init__.py │ │ │ │ ├── clip_wrapper.py │ │ │ │ ├── stable_diffusion.py │ │ │ │ ├── stable_diffusion_inpaint.py │ │ │ │ └── vae.py │ │ │ ├── stable_diffusion_xl/ │ │ │ │ ├── __init__.py │ │ │ │ └── stable_diffusion_xl.py │ │ │ ├── stylegan1/ │ │ │ │ ├── __init__.py │ │ │ │ ├── stylegan1.py │ │ │ │ ├── stylegan1_discriminator.py │ │ │ │ ├── stylegan1_generator.py │ │ │ │ ├── stylegan1_modules.py │ │ │ │ └── stylegan_utils.py │ │ │ ├── stylegan2/ │ │ │ │ ├── __init__.py │ │ │ │ ├── ada/ │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── augment.py │ │ │ │ │ ├── grid_sample_gradfix.py │ │ │ │ │ ├── misc.py │ │ │ │ │ └── upfirdn2d.py │ │ │ │ ├── stylegan2.py │ │ │ │ ├── stylegan2_discriminator.py │ │ │ │ ├── stylegan2_generator.py │ │ │ │ └── stylegan2_modules.py │ │ │ ├── stylegan3/ │ │ │ │ ├── __init__.py │ │ │ │ ├── stylegan3.py │ │ │ │ ├── stylegan3_generator.py │ │ │ │ ├── stylegan3_modules.py │ │ │ │ └── stylegan3_utils.py │ │ │ ├── swinir/ │ │ │ │ ├── __init__.py │ │ │ │ ├── swinir_modules.py │ │ │ │ ├── swinir_net.py │ │ │ │ ├── swinir_rstb.py │ │ │ │ └── swinir_utils.py │ │ │ ├── tdan/ │ │ │ │ ├── __init__.py │ │ │ │ ├── tdan.py │ │ │ │ └── tdan_net.py │ │ │ ├── textual_inversion/ │ │ │ │ ├── __init__.py │ │ │ │ └── textual_inversion.py │ │ │ ├── tof/ │ │ │ │ ├── __init__.py │ │ │ │ ├── tof_vfi_net.py │ │ │ │ └── tof_vsr_net.py │ │ │ ├── ttsr/ │ │ │ │ ├── __init__.py │ │ │ │ ├── lte.py │ │ │ │ ├── search_transformer.py │ │ │ │ ├── ttsr.py │ │ │ │ ├── ttsr_disc.py │ │ │ │ └── ttsr_net.py │ │ │ ├── vico/ │ │ │ │ ├── __init__.py │ │ │ │ ├── vico.py │ │ │ │ └── vico_utils.py │ │ │ └── wgan_gp/ │ │ │ ├── __init__.py │ │ │ ├── wgan_discriminator.py │ │ │ ├── wgan_generator.py │ │ │ ├── wgan_gp.py │ │ │ └── wgan_gp_module.py │ │ ├── losses/ │ │ │ ├── __init__.py │ │ │ ├── adv_loss.py │ │ │ ├── clip_loss.py │ │ │ ├── composition_loss.py │ │ │ ├── face_id_loss.py │ │ │ ├── feature_loss.py │ │ │ ├── gan_loss.py │ │ │ ├── gradient_loss.py │ │ │ ├── loss_comps/ │ │ │ │ ├── __init__.py │ │ │ │ ├── clip_loss_comps.py │ │ │ │ ├── disc_auxiliary_loss_comps.py │ │ │ │ ├── face_id_loss_comps.py │ │ │ │ ├── gan_loss_comps.py │ │ │ │ └── gen_auxiliary_loss_comps.py │ │ │ ├── loss_wrapper.py │ │ │ ├── perceptual_loss.py │ │ │ └── pixelwise_loss.py │ │ └── utils/ │ │ ├── __init__.py │ │ ├── bbox_utils.py │ │ ├── diffusion_utils.py │ │ ├── flow_warp.py │ │ ├── model_utils.py │ │ ├── sampling_utils.py │ │ ├── tensor_utils.py │ │ └── tome_utils.py │ ├── registry.py │ ├── structures/ │ │ ├── __init__.py │ │ └── data_sample.py │ ├── utils/ │ │ ├── __init__.py │ │ ├── cli.py │ │ ├── collect_env.py │ │ ├── img_utils.py │ │ ├── io_utils.py │ │ ├── logger.py │ │ ├── sampler.py │ │ ├── setup_env.py │ │ ├── trans_utils.py │ │ └── typing.py │ ├── version.py │ └── visualization/ │ ├── __init__.py │ ├── concat_visualizer.py │ ├── vis_backend.py │ └── visualizer.py ├── model-index.yml ├── projects/ │ ├── README.md │ ├── animated_drawings/ │ │ ├── README.md │ │ ├── bvh/ │ │ │ └── zombie.bvh │ │ ├── characters/ │ │ │ └── slamdunk/ │ │ │ └── char_cfg.yaml │ │ ├── configs/ │ │ │ ├── motion/ │ │ │ │ └── zombie.yaml │ │ │ ├── mvc/ │ │ │ │ └── slamdunk.yaml │ │ │ └── retarget/ │ │ │ └── fair1_spf.yaml │ │ └── tools/ │ │ ├── generate_character.py │ │ └── generate_video.py │ ├── example_project/ │ │ ├── README.md │ │ ├── configs/ │ │ │ └── examplenet_8xb32_in1k.py │ │ └── models/ │ │ ├── __init__.py │ │ └── example_net.py │ ├── flow_style_vton/ │ │ ├── .gitignore │ │ ├── README.md │ │ ├── configs/ │ │ │ └── flow_style_vton_PFAFN_epoch_101.py │ │ ├── inference.py │ │ ├── models/ │ │ │ ├── __init__.py │ │ │ ├── afwm.py │ │ │ ├── flow_style_vton_model.py │ │ │ └── generator.py │ │ ├── test_pairs.txt │ │ └── vton_dataset.py │ ├── glide/ │ │ ├── configs/ │ │ │ ├── README.md │ │ │ ├── glide_ddim-classifier-free_laion-64-256.py │ │ │ └── glide_ddim-classifier-free_laion-64x64.py │ │ └── models/ │ │ ├── __init__.py │ │ ├── glide.py │ │ ├── glide_modules.py │ │ ├── glide_tokenizer/ │ │ │ ├── __init__.py │ │ │ ├── bpe.py │ │ │ └── simple_tokenizer.py │ │ └── text2im_unet.py │ ├── magicmaker/ │ │ └── index.html │ ├── powerpaint/ │ │ ├── README.md │ │ ├── gradio_PowerPaint.py │ │ ├── pipeline/ │ │ │ ├── pipeline_PowerPaint.py │ │ │ └── pipeline_PowerPaint_ControlNet.py │ │ ├── requirements.txt │ │ └── utils/ │ │ └── utils.py │ └── prompt_to_prompt/ │ ├── README.md │ ├── inversions/ │ │ ├── __init__.py │ │ ├── ddim_inversion.py │ │ └── null_text_inversion.py │ ├── models/ │ │ ├── ptp.py │ │ ├── ptp_utils.py │ │ └── seq_aligner.py │ └── visualize.ipynb ├── requirements/ │ ├── docs.txt │ ├── mminstall.txt │ ├── optional.txt │ ├── readthedocs.txt │ ├── runtime.txt │ └── tests.txt ├── requirements.txt ├── setup.cfg ├── setup.py ├── tests/ │ ├── configs/ │ │ ├── aot_test.py │ │ ├── deepfillv1_test.py │ │ ├── diffuser_wrapper_cfg/ │ │ │ └── config.json │ │ ├── gl_test.py │ │ ├── one_stage_gl.py │ │ ├── pconv_test.py │ │ └── two_stage_test.py │ ├── data/ │ │ ├── coco/ │ │ │ └── annotations/ │ │ │ ├── captions_train2014.json │ │ │ └── captions_val2014.json │ │ ├── controlnet/ │ │ │ └── prompt.json │ │ ├── dataset/ │ │ │ ├── anno.json │ │ │ ├── anno.txt │ │ │ ├── c/ │ │ │ │ └── c.not_support_type │ │ │ ├── classes.txt │ │ │ └── wrong.yml │ │ ├── frames/ │ │ │ ├── ann1.txt │ │ │ ├── ann2.txt │ │ │ └── ann3.txt │ │ ├── image/ │ │ │ └── train.txt │ │ ├── inpainting/ │ │ │ ├── mask_list.txt │ │ │ └── mask_list_single_ch.txt │ │ ├── lq.lmdb/ │ │ │ ├── data.mdb │ │ │ ├── lock.mdb │ │ │ └── meta_info.txt │ │ ├── matting_dataset/ │ │ │ ├── ann.json │ │ │ └── ann_old.json │ │ └── textual_inversion/ │ │ └── imagenet_templates_small.txt │ ├── test_apis/ │ │ ├── test_inferencers/ │ │ │ ├── test_base_mmagic_inferencer.py │ │ │ ├── test_colorization_inferencer.py │ │ │ ├── test_conditional_inferencer.py │ │ │ ├── test_diffusers_pipeline_inferencer.py │ │ │ ├── test_eg3d_inferencer.py │ │ │ ├── test_image_super_resolution_inferencer.py │ │ │ ├── test_inference_functions.py │ │ │ ├── test_inpainting_inferencer.py │ │ │ ├── test_matting_inferencer.py │ │ │ ├── test_mmedit_inferencer.py │ │ │ ├── test_text2image_inferencers.py │ │ │ ├── test_translation_inferencer.py │ │ │ ├── test_unconditional_inferencer.py │ │ │ ├── test_video_interpolation_inferencer.py │ │ │ └── test_video_restoration_inferencer.py │ │ └── test_mmagic_inferencer.py │ ├── test_datasets/ │ │ ├── test_basic_conditional_dataset.py │ │ ├── test_basic_frames_dataset.py │ │ ├── test_basic_image_dataset.py │ │ ├── test_categories.py │ │ ├── test_cifar10_dataset.py │ │ ├── test_comp1k_dataset.py │ │ ├── test_controlnet_dataset.py │ │ ├── test_data_utils.py │ │ ├── test_dreambooth_dataset.py │ │ ├── test_grow_scale_image_dataset.py │ │ ├── test_imagenet_dataset.py │ │ ├── test_mscoco_dataset.py │ │ ├── test_paired_image_dataset.py │ │ ├── test_singan_dataset.py │ │ ├── test_textual_inversion_dataset.py │ │ ├── test_transforms/ │ │ │ ├── test_albumentations.py │ │ │ ├── test_alpha.py │ │ │ ├── test_aug_frames.py │ │ │ ├── test_aug_pixel.py │ │ │ ├── test_aug_shape.py │ │ │ ├── test_blur_kernels.py │ │ │ ├── test_crop.py │ │ │ ├── test_fgbg.py │ │ │ ├── test_formatting.py │ │ │ ├── test_generate_assistant.py │ │ │ ├── test_generate_frame_indices.py │ │ │ ├── test_get_masked_image.py │ │ │ ├── test_loading.py │ │ │ ├── test_matlab_like_resize.py │ │ │ ├── test_normalization.py │ │ │ ├── test_random_degradations.py │ │ │ ├── test_random_down_sampling.py │ │ │ ├── test_trimap.py │ │ │ └── test_values.py │ │ └── test_unpaired_image_dataset.py │ ├── test_engine/ │ │ ├── test_hooks/ │ │ │ ├── test_ema.py │ │ │ ├── test_iter_time_hook.py │ │ │ ├── test_pggan_fetch_data_hook.py │ │ │ ├── test_pickle_data_hook.py │ │ │ ├── test_reduce_lr_scheduler_hook.py │ │ │ └── test_visualization_hook.py │ │ ├── test_optimizers/ │ │ │ ├── test_multi_optimizer_constructor.py │ │ │ ├── test_pggan_optimizer_constructor.py │ │ │ └── test_singan_optimizer_constructor.py │ │ ├── test_runner/ │ │ │ ├── test_log_processor.py │ │ │ ├── test_loop_utils.py │ │ │ └── test_multi_loops.py │ │ └── test_schedulers/ │ │ ├── test_linear_lr_scheduler_with_interval.py │ │ └── test_reduce_lr_scheduler.py │ ├── test_evaluation/ │ │ ├── test_evaluator.py │ │ ├── test_functional/ │ │ │ ├── test_fid_inception.py │ │ │ ├── test_gaussian_funcs.py │ │ │ └── test_inception_utils.py │ │ └── test_metrics/ │ │ ├── test_base_gen_metric.py │ │ ├── test_base_sample_wise_metric.py │ │ ├── test_connectivity_error.py │ │ ├── test_equivariance.py │ │ ├── test_fid.py │ │ ├── test_gradient_error.py │ │ ├── test_inception_score.py │ │ ├── test_mae.py │ │ ├── test_matting_mse.py │ │ ├── test_metrics_utils.py │ │ ├── test_ms_ssim.py │ │ ├── test_mse.py │ │ ├── test_niqe.py │ │ ├── test_ppl.py │ │ ├── test_precision_and_recall.py │ │ ├── test_psnr.py │ │ ├── test_sad.py │ │ ├── test_snr.py │ │ ├── test_ssim.py │ │ └── test_swd.py │ ├── test_models/ │ │ ├── test_archs/ │ │ │ ├── test_aspp.py │ │ │ ├── test_conv.py │ │ │ ├── test_downsample.py │ │ │ ├── test_ensemble.py │ │ │ ├── test_gated_conv_module.py │ │ │ ├── test_img_normalize.py │ │ │ ├── test_linear_module.py │ │ │ ├── test_lora.py │ │ │ ├── test_multi_layer_disc.py │ │ │ ├── test_patch_disc.py │ │ │ ├── test_resnet.py │ │ │ ├── test_separable_conv_module.py │ │ │ ├── test_simple_encoder_decoder.py │ │ │ ├── test_smpatch_disc.py │ │ │ ├── test_sr_backbone.py │ │ │ ├── test_tokenizer.py │ │ │ ├── test_upsample.py │ │ │ ├── test_vgg.py │ │ │ └── test_wrapper.py │ │ ├── test_base_models/ │ │ │ ├── test_average_model.py │ │ │ ├── test_base_conditional_gan.py │ │ │ ├── test_base_edit_model.py │ │ │ ├── test_base_gan.py │ │ │ ├── test_base_mattor.py │ │ │ ├── test_base_translation_model.py │ │ │ ├── test_basic_interpolator.py │ │ │ ├── test_one_stage.py │ │ │ └── test_two_stage.py │ │ ├── test_data_preprocessors/ │ │ │ ├── test_data_preprocessor.py │ │ │ └── test_mattor_preprocessor.py │ │ ├── test_diffusion_schedulers/ │ │ │ ├── test_ddim_scheduler.py │ │ │ └── test_init.py │ │ ├── test_editors/ │ │ │ ├── test_animatediff/ │ │ │ │ ├── test_animatediff.py │ │ │ │ ├── test_attention3d.py │ │ │ │ ├── test_motion_module.py │ │ │ │ ├── test_res_blocks3d.py │ │ │ │ ├── test_unet3d.py │ │ │ │ └── test_unet_blocks3d.py │ │ │ ├── test_aotgan/ │ │ │ │ ├── test_aot_decoder.py │ │ │ │ ├── test_aot_encoder.py │ │ │ │ ├── test_aot_encoder_decoder.py │ │ │ │ ├── test_aot_inpaintor.py │ │ │ │ └── test_aot_neck.py │ │ │ ├── test_arcface/ │ │ │ │ ├── test_arcface_modules.py │ │ │ │ ├── test_id_loss.py │ │ │ │ └── test_model_irse.py │ │ │ ├── test_basicvsr/ │ │ │ │ ├── test_basicvsr.py │ │ │ │ └── test_basicvsr_net.py │ │ │ ├── test_basicvsr_plusplus_net/ │ │ │ │ └── test_basicvsr_plusplus_net.py │ │ │ ├── test_biggan/ │ │ │ │ ├── test_biggan.py │ │ │ │ ├── test_biggan_deep_discriminator.py │ │ │ │ ├── test_biggan_deep_generator.py │ │ │ │ ├── test_biggan_discriminator.py │ │ │ │ ├── test_biggan_generator.py │ │ │ │ ├── test_biggan_modules.py │ │ │ │ └── test_biggan_snmodule.py │ │ │ ├── test_cain/ │ │ │ │ ├── test_cain.py │ │ │ │ └── test_cain_net.py │ │ │ ├── test_controlnet/ │ │ │ │ ├── test_controlnet.py │ │ │ │ └── test_controlnet_utils.py │ │ │ ├── test_cyclegan/ │ │ │ │ ├── test_cyclegan.py │ │ │ │ ├── test_cyclegan_generator.py │ │ │ │ └── test_cyclegan_modules.py │ │ │ ├── test_dcgan/ │ │ │ │ ├── test_dcgan.py │ │ │ │ ├── test_dcgan_discriminator.py │ │ │ │ └── test_dcgan_generator.py │ │ │ ├── test_ddpm/ │ │ │ │ ├── test_attention.py │ │ │ │ ├── test_ddpm_scheduler.py │ │ │ │ ├── test_denoising_unet.py │ │ │ │ ├── test_embeddings.py │ │ │ │ ├── test_res_blocks.py │ │ │ │ └── test_unet_blocks.py │ │ │ ├── test_deblurganv2/ │ │ │ │ ├── test_deblurganv2.py │ │ │ │ ├── test_deblurganv2_discriminator.py │ │ │ │ └── test_deblurganv2_generator.py │ │ │ ├── test_deepfillv1/ │ │ │ │ ├── test_contextual_attention.py │ │ │ │ ├── test_contextual_attention_neck.py │ │ │ │ ├── test_deepfill_decoder.py │ │ │ │ ├── test_deepfill_disc.py │ │ │ │ ├── test_deepfill_encoder.py │ │ │ │ ├── test_deepfill_refiner.py │ │ │ │ └── test_deepfillv1.py │ │ │ ├── test_deepfillv2/ │ │ │ │ └── test_two_stage_encoder_decoder.py │ │ │ ├── test_dic/ │ │ │ │ ├── test_dic.py │ │ │ │ ├── test_dic_net.py │ │ │ │ ├── test_feedback_hour_glass.py │ │ │ │ └── test_light_cnn.py │ │ │ ├── test_dim/ │ │ │ │ └── test_dim.py │ │ │ ├── test_disco_diffusion/ │ │ │ │ ├── test_disco_diffusion.py │ │ │ │ └── test_disco_diffusion_clip_wrapper.py │ │ │ ├── test_dreambooth/ │ │ │ │ └── test_dreambooth.py │ │ │ ├── test_duf/ │ │ │ │ └── test_duf.py │ │ │ ├── test_edsr/ │ │ │ │ └── test_edsr_net.py │ │ │ ├── test_edvr/ │ │ │ │ ├── test_edvr.py │ │ │ │ └── test_edvr_net.py │ │ │ ├── test_eg3d/ │ │ │ │ ├── test_camera.py │ │ │ │ ├── test_dual_discriminator.py │ │ │ │ ├── test_eg3d.py │ │ │ │ ├── test_eg3d_generator.py │ │ │ │ ├── test_eg3d_modules.py │ │ │ │ ├── test_eg3d_utils.py │ │ │ │ ├── test_ray_sampler.py │ │ │ │ └── test_renderer.py │ │ │ ├── test_esrgan/ │ │ │ │ ├── test_esrgan.py │ │ │ │ └── test_rrdb_net.py │ │ │ ├── test_fastcomposer/ │ │ │ │ └── test_fastcomposer.py │ │ │ ├── test_fba/ │ │ │ │ ├── test_fba_decoder.py │ │ │ │ └── test_fba_encoder.py │ │ │ ├── test_flavr/ │ │ │ │ ├── test_flavr.py │ │ │ │ └── test_flavr_net.py │ │ │ ├── test_gca/ │ │ │ │ ├── test_gca.py │ │ │ │ ├── test_gca_module.py │ │ │ │ ├── test_resgca_dec.py │ │ │ │ └── test_resgca_enc.py │ │ │ ├── test_ggan/ │ │ │ │ └── test_ggan.py │ │ │ ├── test_glean/ │ │ │ │ └── test_glean_styleganv2.py │ │ │ ├── test_glide/ │ │ │ │ └── test_glide.py │ │ │ ├── test_global_local/ │ │ │ │ ├── test_gl_decoder.py │ │ │ │ ├── test_gl_dilation.py │ │ │ │ ├── test_gl_disc.py │ │ │ │ ├── test_gl_encoder.py │ │ │ │ ├── test_gl_encoder_decoder.py │ │ │ │ └── test_gl_inpaintor.py │ │ │ ├── test_guided_diffusion/ │ │ │ │ └── test_adm.py │ │ │ ├── test_iconvsr/ │ │ │ │ └── test_iconvsr_net.py │ │ │ ├── test_indexnet/ │ │ │ │ ├── test_indexnet.py │ │ │ │ ├── test_indexnet_decoder.py │ │ │ │ └── test_indexnet_encoder.py │ │ │ ├── test_inst_colorization/ │ │ │ │ ├── test_color_utils.py │ │ │ │ ├── test_colorization_net.py │ │ │ │ ├── test_fusion_net.py │ │ │ │ ├── test_inst_colorization.py │ │ │ │ └── test_weight_layer.py │ │ │ ├── test_liif/ │ │ │ │ ├── test_liif.py │ │ │ │ ├── test_liif_net.py │ │ │ │ └── test_mlp_refiner.py │ │ │ ├── test_lsgan/ │ │ │ │ ├── test_lsgan.py │ │ │ │ ├── test_lsgan_discriminator.py │ │ │ │ └── test_lsgan_generator.py │ │ │ ├── test_mspie/ │ │ │ │ ├── test_mspie_stylegan2.py │ │ │ │ ├── test_mspie_stylegan2_discriminator.py │ │ │ │ ├── test_mspie_stylegan2_generator.py │ │ │ │ ├── test_mspie_stylegan2_modules.py │ │ │ │ ├── test_pe_singan.py │ │ │ │ ├── test_pe_singan_generator.py │ │ │ │ └── test_positional_encoding.py │ │ │ ├── test_nafnet/ │ │ │ │ ├── test_naf_avgpool2d.py │ │ │ │ ├── test_naf_layernorm2d.py │ │ │ │ ├── test_nafbaseline.py │ │ │ │ └── test_nafnet.py │ │ │ ├── test_pconv/ │ │ │ │ ├── test_mask_conv_module.py │ │ │ │ ├── test_partial_conv.py │ │ │ │ ├── test_pconv_decoder.py │ │ │ │ ├── test_pconv_encoder.py │ │ │ │ ├── test_pconv_encoder_decoder.py │ │ │ │ └── test_pconv_inpaintor.py │ │ │ ├── test_pggan/ │ │ │ │ ├── test_pggan.py │ │ │ │ ├── test_pggan_discriminator.py │ │ │ │ ├── test_pggan_generator.py │ │ │ │ └── test_pggan_modules.py │ │ │ ├── test_pix2pix/ │ │ │ │ ├── test_pix2pix.py │ │ │ │ ├── test_pix2pix_generator.py │ │ │ │ └── test_pix2pix_modules.py │ │ │ ├── test_plain/ │ │ │ │ ├── test_plain_decoder.py │ │ │ │ └── test_plain_refiner.py │ │ │ ├── test_rdn/ │ │ │ │ └── test_rdn_net.py │ │ │ ├── test_real_basicvsr/ │ │ │ │ ├── test_real_basicvsr.py │ │ │ │ └── test_real_basicvsr_net.py │ │ │ ├── test_real_esrgan/ │ │ │ │ ├── test_real_esrgan.py │ │ │ │ └── test_unet_disc.py │ │ │ ├── test_restormer/ │ │ │ │ └── test_restormer_net.py │ │ │ ├── test_sagan/ │ │ │ │ ├── test_sagan.py │ │ │ │ ├── test_sagan_discriminator.py │ │ │ │ ├── test_sagan_generator.py │ │ │ │ └── test_sagan_modules.py │ │ │ ├── test_singan/ │ │ │ │ ├── test_singan.py │ │ │ │ ├── test_singan_discriminator.py │ │ │ │ ├── test_singan_generator.py │ │ │ │ └── test_singan_modules.py │ │ │ ├── test_srcnn/ │ │ │ │ └── test_srcnn_net.py │ │ │ ├── test_srgan/ │ │ │ │ ├── test_modified_vgg.py │ │ │ │ ├── test_sr_resnet.py │ │ │ │ └── test_srgan.py │ │ │ ├── test_stable_diffusion/ │ │ │ │ ├── test_clip_wrapper.py │ │ │ │ ├── test_stable_diffusion.py │ │ │ │ ├── test_stable_diffusion_inpaint.py │ │ │ │ ├── test_stable_diffusion_tomesd.py │ │ │ │ └── test_vae.py │ │ │ ├── test_stable_diffusion_xl/ │ │ │ │ └── test_stable_diffusion_xl.py │ │ │ ├── test_stylegan1/ │ │ │ │ ├── test_stylegan1.py │ │ │ │ ├── test_stylegan1_discriminator.py │ │ │ │ ├── test_stylegan1_generator.py │ │ │ │ ├── test_stylegan1_modules.py │ │ │ │ └── test_stylegan_utils.py │ │ │ ├── test_stylegan2/ │ │ │ │ ├── test_ada/ │ │ │ │ │ └── test_augment.py │ │ │ │ ├── test_stylegan2.py │ │ │ │ ├── test_stylegan2_discriminator.py │ │ │ │ ├── test_stylegan2_generator.py │ │ │ │ └── test_stylegan2_modules.py │ │ │ ├── test_stylegan3/ │ │ │ │ ├── test_stylegan3.py │ │ │ │ ├── test_stylegan3_generator.py │ │ │ │ ├── test_stylegan3_modules.py │ │ │ │ └── test_stylegan3_utils.py │ │ │ ├── test_swinir/ │ │ │ │ ├── test_swinir_modules.py │ │ │ │ ├── test_swinir_net.py │ │ │ │ ├── test_swinir_rstb.py │ │ │ │ └── test_swinir_utils.py │ │ │ ├── test_tdan/ │ │ │ │ ├── test_tdan.py │ │ │ │ └── test_tdan_net.py │ │ │ ├── test_tof/ │ │ │ │ ├── test_tof_vfi_net.py │ │ │ │ └── test_tof_vsr_net.py │ │ │ ├── test_ttsr/ │ │ │ │ ├── test_lte.py │ │ │ │ ├── test_search_transformer.py │ │ │ │ ├── test_ttsr.py │ │ │ │ ├── test_ttsr_disc.py │ │ │ │ └── test_ttsr_net.py │ │ │ ├── test_vico/ │ │ │ │ ├── test_vico.py │ │ │ │ └── test_vico_utils.py │ │ │ └── test_wgan_gp/ │ │ │ ├── test_wgan_discriminator.py │ │ │ ├── test_wgan_generator.py │ │ │ ├── test_wgan_gp.py │ │ │ └── test_wgan_gp_module.py │ │ ├── test_losses/ │ │ │ ├── test_clip_loss.py │ │ │ ├── test_composition_loss.py │ │ │ ├── test_face_id_loss.py │ │ │ ├── test_feature_loss.py │ │ │ ├── test_gan_loss.py │ │ │ ├── test_gradient_loss.py │ │ │ ├── test_loss_comps/ │ │ │ │ ├── test_clip_loss_comps.py │ │ │ │ ├── test_disc_auxiliary_loss_comps.py │ │ │ │ ├── test_face_id_loss_comps.py │ │ │ │ ├── test_gan_loss_comps.py │ │ │ │ └── test_gen_auxiliary_loss_comps.py │ │ │ ├── test_loss_wrapper.py │ │ │ ├── test_perceptual_loss.py │ │ │ └── test_pixelwise_loss.py │ │ └── test_utils/ │ │ ├── test_bbox_utils.py │ │ ├── test_flow_warp.py │ │ ├── test_model_utils.py │ │ ├── test_sampling_utils.py │ │ └── test_tensor_utils.py │ ├── test_structures/ │ │ └── test_data_sample.py │ ├── test_utils/ │ │ ├── test_cli.py │ │ ├── test_img_utils.py │ │ ├── test_io_utils.py │ │ ├── test_logger.py │ │ ├── test_sampler.py │ │ ├── test_setup_env.py │ │ └── test_trans_utils.py │ └── test_visualization/ │ ├── test_concat_visualizer.py │ ├── test_vis_backend.py │ └── test_visualizer.py └── tools/ ├── analysis_tools/ │ ├── get_flops.py │ └── print_config.py ├── cpu_train.sh ├── dataset_converters/ │ ├── bgm/ │ │ └── preprocess_bgm_dataset.py │ ├── celeba-hq/ │ │ ├── README.md │ │ └── README_zh-CN.md │ ├── classic5/ │ │ ├── README.md │ │ └── README_zh-CN.md │ ├── comp1k/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── check_extended_fg.py │ │ ├── evaluate_comp1k.py │ │ ├── extend_fg.py │ │ ├── filter_comp1k_anno.py │ │ └── preprocess_comp1k_dataset.py │ ├── denoising/ │ │ ├── README.md │ │ └── README_zh-CN.md │ ├── deraining/ │ │ ├── README.md │ │ └── README_zh-CN.md │ ├── df2k_ost/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ └── preprocess_df2k_ost_dataset.py │ ├── div2k/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ └── preprocess_div2k_dataset.py │ ├── dpdd/ │ │ ├── README.md │ │ └── README_zh-CN.md │ ├── glean/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── preprocess_cat_test_dataset.py │ │ ├── preprocess_cat_train_dataset.py │ │ └── preprocess_ffhq_celebahq_dataset.py │ ├── gopro/ │ │ ├── README.md │ │ └── README_zh-CN.md │ ├── hide/ │ │ ├── README.md │ │ └── README_zh-CN.md │ ├── live1/ │ │ ├── README.md │ │ └── README_zh-CN.md │ ├── ntire21_decompression/ │ │ ├── README.md │ │ └── README_zh-CN.md │ ├── paired-pix2pix/ │ │ ├── README.md │ │ └── README_zh-CN.md │ ├── paris-street-view/ │ │ ├── README.md │ │ └── README_zh-CN.md │ ├── places365/ │ │ ├── README.md │ │ └── README_zh-CN.md │ ├── realblur/ │ │ ├── README.md │ │ └── README_zh-CN.md │ ├── realsrset/ │ │ ├── README.md │ │ └── README_zh-CN.md │ ├── reds/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ ├── crop_sub_images.py │ │ └── preprocess_reds_dataset.py │ ├── sidd/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ └── preprocess_sidd_test_dataset.py │ ├── spmcs/ │ │ ├── README.md │ │ └── README_zh-CN.md │ ├── udm10/ │ │ ├── README.md │ │ └── README_zh-CN.md │ ├── unconditional_gans/ │ │ └── README.md │ ├── unpaired-cyclegan/ │ │ ├── README.md │ │ └── README_zh-CN.md │ ├── vid4/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ └── preprocess_vid4_dataset.py │ ├── videolq/ │ │ ├── README.md │ │ └── README_zh-CN.md │ ├── vimeo90k/ │ │ ├── README.md │ │ ├── README_zh-CN.md │ │ └── preprocess_vimeo90k_dataset.py │ └── vimeo90k-triplet/ │ ├── README.md │ └── README_zh-CN.md ├── dist_test.sh ├── dist_train.sh ├── gui/ │ ├── README.md │ ├── component.py │ ├── gui.py │ ├── page_general.py │ ├── page_sr.py │ └── utils.py ├── model_converters/ │ └── publish_model.py ├── slurm_test.sh ├── slurm_train.sh ├── test.py └── train.py