├── LICENSE.txt ├── README.md ├── basicsr ├── __init__.py ├── archs │ ├── __init__.py │ ├── arch_util.py │ ├── basicvsr_arch.py │ ├── basicvsrpp_arch.py │ ├── degradat_arch.py │ ├── dfdnet_arch.py │ ├── dfdnet_util.py │ ├── discriminator_arch.py │ ├── duf_arch.py │ ├── ecbsr_arch.py │ ├── edsr_arch.py │ ├── edvr_arch.py │ ├── hifacegan_arch.py │ ├── hifacegan_util.py │ ├── inception.py │ ├── rcan_arch.py │ ├── ridnet_arch.py │ ├── rrdbnet_arch.py │ ├── spynet_arch.py │ ├── srresnet_arch.py │ ├── srvgg_arch.py │ ├── stylegan2_arch.py │ ├── stylegan2_bilinear_arch.py │ ├── swinir_arch.py │ ├── tof_arch.py │ └── vgg_arch.py ├── data │ ├── __init__.py │ ├── data_sampler.py │ ├── data_util.py │ ├── degradations.py │ ├── ffhq_dataset.py │ ├── ffhq_degradation_dataset.py │ ├── paired_image_dataset.py │ ├── prefetch_dataloader.py │ ├── realesrgan_dataset.py │ ├── realesrgan_paired_dataset.py │ ├── reds_dataset.py │ ├── single_image_dataset.py │ ├── transforms.py │ ├── video_test_dataset.py │ └── vimeo90k_dataset.py ├── losses │ ├── __init__.py │ ├── basic_loss.py │ ├── gan_loss.py │ └── loss_util.py ├── metrics │ ├── README.md │ ├── README_CN.md │ ├── __init__.py │ ├── fid.py │ ├── metric_util.py │ ├── niqe.py │ ├── niqe_pris_params.npz │ ├── psnr_ssim.py │ └── test_metrics │ │ └── test_psnr_ssim.py ├── models │ ├── __init__.py │ ├── base_model.py │ ├── edvr_model.py │ ├── esrgan_model.py │ ├── hifacegan_model.py │ ├── lr_scheduler.py │ ├── realesrgan_model.py │ ├── realesrnet_model.py │ ├── sr_model.py │ ├── srgan_model.py │ ├── stylegan2_model.py │ ├── swinir_model.py │ ├── video_base_model.py │ ├── video_gan_model.py │ ├── video_recurrent_gan_model.py │ └── video_recurrent_model.py ├── ops │ ├── __init__.py │ ├── dcn │ │ ├── __init__.py │ │ └── deform_conv.py │ ├── fused_act │ │ ├── __init__.py │ │ └── fused_act.py │ └── upfirdn2d │ │ ├── __init__.py │ │ └── upfirdn2d.py ├── test.py ├── train.py └── utils │ ├── __init__.py │ ├── color_util.py │ ├── diffjpeg.py │ ├── dist_util.py │ ├── download_util.py │ ├── file_client.py │ ├── flow_util.py │ ├── img_process_util.py │ ├── img_util.py │ ├── lmdb_util.py │ ├── logger.py │ ├── matlab_functions.py │ ├── misc.py │ ├── options.py │ ├── plot_util.py │ ├── realesrgan_utils.py │ └── registry.py ├── configs ├── CoSeR │ ├── aia_512_imagenet_all_caption_clip_atten_ref.yaml │ ├── inference.yaml │ └── qformer_srresnet_imagenet_all.yaml └── autoencoder │ └── autoencoder_kl_64x64x4_resi.yaml ├── data ├── ImageNet │ └── class.txt ├── count_clip_sim.py ├── generate_caption.py └── prepare_imagenet.py ├── images └── teaser.jpg ├── ldm ├── data │ ├── __init__.py │ ├── base.py │ ├── imagenet.py │ └── lsun.py ├── lr_scheduler.py ├── models │ ├── Qformer.py │ ├── autoencoder.py │ ├── diffusion │ │ ├── __init__.py │ │ ├── classifier.py │ │ ├── ddim.py │ │ ├── ddpm.py │ │ ├── ddpm_inv.py │ │ └── plms.py │ ├── global_adapter.py │ └── respace.py ├── modules │ ├── attention.py │ ├── diffusionmodules │ │ ├── __init__.py │ │ ├── model.py │ │ ├── openaimodel.py │ │ └── util.py │ ├── distributions │ │ ├── __init__.py │ │ └── distributions.py │ ├── ema.py │ ├── embedding_manager.py │ ├── encoders │ │ ├── __init__.py │ │ ├── modules.py │ │ └── transformer_utils.py │ ├── image_degradation │ │ ├── __init__.py │ │ ├── bsrgan.py │ │ ├── bsrgan_light.py │ │ ├── utils │ │ │ └── test.png │ │ └── utils_image.py │ ├── losses │ │ ├── __init__.py │ │ ├── contperceptual.py │ │ └── vqperceptual.py │ ├── spade.py │ ├── swinir.py │ └── x_transformer.py └── util.py ├── main.py ├── requirements.txt ├── scripts ├── inference.py ├── inference_tile.py ├── util_image.py └── wavelet_color_fix.py └── setup.py /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/README.md -------------------------------------------------------------------------------- /basicsr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/__init__.py -------------------------------------------------------------------------------- /basicsr/archs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/__init__.py -------------------------------------------------------------------------------- /basicsr/archs/arch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/arch_util.py -------------------------------------------------------------------------------- /basicsr/archs/basicvsr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/basicvsr_arch.py -------------------------------------------------------------------------------- /basicsr/archs/basicvsrpp_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/basicvsrpp_arch.py -------------------------------------------------------------------------------- /basicsr/archs/degradat_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/degradat_arch.py -------------------------------------------------------------------------------- /basicsr/archs/dfdnet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/dfdnet_arch.py -------------------------------------------------------------------------------- /basicsr/archs/dfdnet_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/dfdnet_util.py -------------------------------------------------------------------------------- /basicsr/archs/discriminator_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/discriminator_arch.py -------------------------------------------------------------------------------- /basicsr/archs/duf_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/duf_arch.py -------------------------------------------------------------------------------- /basicsr/archs/ecbsr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/ecbsr_arch.py -------------------------------------------------------------------------------- /basicsr/archs/edsr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/edsr_arch.py -------------------------------------------------------------------------------- /basicsr/archs/edvr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/edvr_arch.py -------------------------------------------------------------------------------- /basicsr/archs/hifacegan_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/hifacegan_arch.py -------------------------------------------------------------------------------- /basicsr/archs/hifacegan_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/hifacegan_util.py -------------------------------------------------------------------------------- /basicsr/archs/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/inception.py -------------------------------------------------------------------------------- /basicsr/archs/rcan_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/rcan_arch.py -------------------------------------------------------------------------------- /basicsr/archs/ridnet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/ridnet_arch.py -------------------------------------------------------------------------------- /basicsr/archs/rrdbnet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/rrdbnet_arch.py -------------------------------------------------------------------------------- /basicsr/archs/spynet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/spynet_arch.py -------------------------------------------------------------------------------- /basicsr/archs/srresnet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/srresnet_arch.py -------------------------------------------------------------------------------- /basicsr/archs/srvgg_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/srvgg_arch.py -------------------------------------------------------------------------------- /basicsr/archs/stylegan2_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/stylegan2_arch.py -------------------------------------------------------------------------------- /basicsr/archs/stylegan2_bilinear_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/stylegan2_bilinear_arch.py -------------------------------------------------------------------------------- /basicsr/archs/swinir_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/swinir_arch.py -------------------------------------------------------------------------------- /basicsr/archs/tof_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/tof_arch.py -------------------------------------------------------------------------------- /basicsr/archs/vgg_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/archs/vgg_arch.py -------------------------------------------------------------------------------- /basicsr/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/data/__init__.py -------------------------------------------------------------------------------- /basicsr/data/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/data/data_sampler.py -------------------------------------------------------------------------------- /basicsr/data/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/data/data_util.py -------------------------------------------------------------------------------- /basicsr/data/degradations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/data/degradations.py -------------------------------------------------------------------------------- /basicsr/data/ffhq_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/data/ffhq_dataset.py -------------------------------------------------------------------------------- /basicsr/data/ffhq_degradation_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/data/ffhq_degradation_dataset.py -------------------------------------------------------------------------------- /basicsr/data/paired_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/data/paired_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/prefetch_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/data/prefetch_dataloader.py -------------------------------------------------------------------------------- /basicsr/data/realesrgan_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/data/realesrgan_dataset.py -------------------------------------------------------------------------------- /basicsr/data/realesrgan_paired_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/data/realesrgan_paired_dataset.py -------------------------------------------------------------------------------- /basicsr/data/reds_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/data/reds_dataset.py -------------------------------------------------------------------------------- /basicsr/data/single_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/data/single_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/data/transforms.py -------------------------------------------------------------------------------- /basicsr/data/video_test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/data/video_test_dataset.py -------------------------------------------------------------------------------- /basicsr/data/vimeo90k_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/data/vimeo90k_dataset.py -------------------------------------------------------------------------------- /basicsr/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/losses/__init__.py -------------------------------------------------------------------------------- /basicsr/losses/basic_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/losses/basic_loss.py -------------------------------------------------------------------------------- /basicsr/losses/gan_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/losses/gan_loss.py -------------------------------------------------------------------------------- /basicsr/losses/loss_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/losses/loss_util.py -------------------------------------------------------------------------------- /basicsr/metrics/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/metrics/README.md -------------------------------------------------------------------------------- /basicsr/metrics/README_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/metrics/README_CN.md -------------------------------------------------------------------------------- /basicsr/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/metrics/__init__.py -------------------------------------------------------------------------------- /basicsr/metrics/fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/metrics/fid.py -------------------------------------------------------------------------------- /basicsr/metrics/metric_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/metrics/metric_util.py -------------------------------------------------------------------------------- /basicsr/metrics/niqe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/metrics/niqe.py -------------------------------------------------------------------------------- /basicsr/metrics/niqe_pris_params.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/metrics/niqe_pris_params.npz -------------------------------------------------------------------------------- /basicsr/metrics/psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/metrics/psnr_ssim.py -------------------------------------------------------------------------------- /basicsr/metrics/test_metrics/test_psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/metrics/test_metrics/test_psnr_ssim.py -------------------------------------------------------------------------------- /basicsr/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/models/__init__.py -------------------------------------------------------------------------------- /basicsr/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/models/base_model.py -------------------------------------------------------------------------------- /basicsr/models/edvr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/models/edvr_model.py -------------------------------------------------------------------------------- /basicsr/models/esrgan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/models/esrgan_model.py -------------------------------------------------------------------------------- /basicsr/models/hifacegan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/models/hifacegan_model.py -------------------------------------------------------------------------------- /basicsr/models/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/models/lr_scheduler.py -------------------------------------------------------------------------------- /basicsr/models/realesrgan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/models/realesrgan_model.py -------------------------------------------------------------------------------- /basicsr/models/realesrnet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/models/realesrnet_model.py -------------------------------------------------------------------------------- /basicsr/models/sr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/models/sr_model.py -------------------------------------------------------------------------------- /basicsr/models/srgan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/models/srgan_model.py -------------------------------------------------------------------------------- /basicsr/models/stylegan2_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/models/stylegan2_model.py -------------------------------------------------------------------------------- /basicsr/models/swinir_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/models/swinir_model.py -------------------------------------------------------------------------------- /basicsr/models/video_base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/models/video_base_model.py -------------------------------------------------------------------------------- /basicsr/models/video_gan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/models/video_gan_model.py -------------------------------------------------------------------------------- /basicsr/models/video_recurrent_gan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/models/video_recurrent_gan_model.py -------------------------------------------------------------------------------- /basicsr/models/video_recurrent_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/models/video_recurrent_model.py -------------------------------------------------------------------------------- /basicsr/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /basicsr/ops/dcn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/ops/dcn/__init__.py -------------------------------------------------------------------------------- /basicsr/ops/dcn/deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/ops/dcn/deform_conv.py -------------------------------------------------------------------------------- /basicsr/ops/fused_act/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/ops/fused_act/__init__.py -------------------------------------------------------------------------------- /basicsr/ops/fused_act/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/ops/fused_act/fused_act.py -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/ops/upfirdn2d/__init__.py -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/ops/upfirdn2d/upfirdn2d.py -------------------------------------------------------------------------------- /basicsr/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/test.py -------------------------------------------------------------------------------- /basicsr/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/train.py -------------------------------------------------------------------------------- /basicsr/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/utils/__init__.py -------------------------------------------------------------------------------- /basicsr/utils/color_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/utils/color_util.py -------------------------------------------------------------------------------- /basicsr/utils/diffjpeg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/utils/diffjpeg.py -------------------------------------------------------------------------------- /basicsr/utils/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/utils/dist_util.py -------------------------------------------------------------------------------- /basicsr/utils/download_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/utils/download_util.py -------------------------------------------------------------------------------- /basicsr/utils/file_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/utils/file_client.py -------------------------------------------------------------------------------- /basicsr/utils/flow_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/utils/flow_util.py -------------------------------------------------------------------------------- /basicsr/utils/img_process_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/utils/img_process_util.py -------------------------------------------------------------------------------- /basicsr/utils/img_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/utils/img_util.py -------------------------------------------------------------------------------- /basicsr/utils/lmdb_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/utils/lmdb_util.py -------------------------------------------------------------------------------- /basicsr/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/utils/logger.py -------------------------------------------------------------------------------- /basicsr/utils/matlab_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/utils/matlab_functions.py -------------------------------------------------------------------------------- /basicsr/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/utils/misc.py -------------------------------------------------------------------------------- /basicsr/utils/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/utils/options.py -------------------------------------------------------------------------------- /basicsr/utils/plot_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/utils/plot_util.py -------------------------------------------------------------------------------- /basicsr/utils/realesrgan_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/utils/realesrgan_utils.py -------------------------------------------------------------------------------- /basicsr/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/basicsr/utils/registry.py -------------------------------------------------------------------------------- /configs/CoSeR/aia_512_imagenet_all_caption_clip_atten_ref.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/configs/CoSeR/aia_512_imagenet_all_caption_clip_atten_ref.yaml -------------------------------------------------------------------------------- /configs/CoSeR/inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/configs/CoSeR/inference.yaml -------------------------------------------------------------------------------- /configs/CoSeR/qformer_srresnet_imagenet_all.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/configs/CoSeR/qformer_srresnet_imagenet_all.yaml -------------------------------------------------------------------------------- /configs/autoencoder/autoencoder_kl_64x64x4_resi.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/configs/autoencoder/autoencoder_kl_64x64x4_resi.yaml -------------------------------------------------------------------------------- /data/ImageNet/class.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/data/ImageNet/class.txt -------------------------------------------------------------------------------- /data/count_clip_sim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/data/count_clip_sim.py -------------------------------------------------------------------------------- /data/generate_caption.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/data/generate_caption.py -------------------------------------------------------------------------------- /data/prepare_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/data/prepare_imagenet.py -------------------------------------------------------------------------------- /images/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/images/teaser.jpg -------------------------------------------------------------------------------- /ldm/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/data/base.py -------------------------------------------------------------------------------- /ldm/data/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/data/imagenet.py -------------------------------------------------------------------------------- /ldm/data/lsun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/data/lsun.py -------------------------------------------------------------------------------- /ldm/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/lr_scheduler.py -------------------------------------------------------------------------------- /ldm/models/Qformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/models/Qformer.py -------------------------------------------------------------------------------- /ldm/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/models/autoencoder.py -------------------------------------------------------------------------------- /ldm/models/diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/models/diffusion/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/models/diffusion/classifier.py -------------------------------------------------------------------------------- /ldm/models/diffusion/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/models/diffusion/ddim.py -------------------------------------------------------------------------------- /ldm/models/diffusion/ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/models/diffusion/ddpm.py -------------------------------------------------------------------------------- /ldm/models/diffusion/ddpm_inv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/models/diffusion/ddpm_inv.py -------------------------------------------------------------------------------- /ldm/models/diffusion/plms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/models/diffusion/plms.py -------------------------------------------------------------------------------- /ldm/models/global_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/models/global_adapter.py -------------------------------------------------------------------------------- /ldm/models/respace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/models/respace.py -------------------------------------------------------------------------------- /ldm/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/modules/attention.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/openaimodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/modules/diffusionmodules/openaimodel.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/modules/diffusionmodules/util.py -------------------------------------------------------------------------------- /ldm/modules/distributions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/distributions/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/modules/distributions/distributions.py -------------------------------------------------------------------------------- /ldm/modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/modules/ema.py -------------------------------------------------------------------------------- /ldm/modules/embedding_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/modules/embedding_manager.py -------------------------------------------------------------------------------- /ldm/modules/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/encoders/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/modules/encoders/modules.py -------------------------------------------------------------------------------- /ldm/modules/encoders/transformer_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/modules/encoders/transformer_utils.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/modules/image_degradation/__init__.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/bsrgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/modules/image_degradation/bsrgan.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/bsrgan_light.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/modules/image_degradation/bsrgan_light.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/utils/test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/modules/image_degradation/utils/test.png -------------------------------------------------------------------------------- /ldm/modules/image_degradation/utils_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/modules/image_degradation/utils_image.py -------------------------------------------------------------------------------- /ldm/modules/losses/__init__.py: -------------------------------------------------------------------------------- 1 | from ldm.modules.losses.contperceptual import LPIPSWithDiscriminator -------------------------------------------------------------------------------- /ldm/modules/losses/contperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/modules/losses/contperceptual.py -------------------------------------------------------------------------------- /ldm/modules/losses/vqperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/modules/losses/vqperceptual.py -------------------------------------------------------------------------------- /ldm/modules/spade.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/modules/spade.py -------------------------------------------------------------------------------- /ldm/modules/swinir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/modules/swinir.py -------------------------------------------------------------------------------- /ldm/modules/x_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/modules/x_transformer.py -------------------------------------------------------------------------------- /ldm/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/ldm/util.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/main.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/scripts/inference.py -------------------------------------------------------------------------------- /scripts/inference_tile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/scripts/inference_tile.py -------------------------------------------------------------------------------- /scripts/util_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/scripts/util_image.py -------------------------------------------------------------------------------- /scripts/wavelet_color_fix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/scripts/wavelet_color_fix.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TianheWu/CoSeR/HEAD/setup.py --------------------------------------------------------------------------------