├── .gitignore ├── LICENSE ├── README.md ├── _configs ├── algo │ ├── ddim.yaml │ ├── ddrm.yaml │ ├── dps.yaml │ ├── identity.yaml │ ├── mcg.yaml │ ├── pgdm.yaml │ ├── reddiff.yaml │ ├── reddiff_parallel.yaml │ ├── sds.yaml │ └── sds_var.yaml ├── ca.yaml ├── classifier │ ├── imagenet256_cond.yaml │ ├── imagenet512_cond.yaml │ └── none.yaml ├── dataset │ ├── ffhq256_train.yaml │ ├── ffhq256_val.yaml │ ├── imagenet256_train.yaml │ ├── imagenet256_val.yaml │ └── imagenet512_val.yaml ├── ddrmpp.yaml ├── ddrmpp_ffhq.yaml ├── diffusion │ └── linear1000.yaml ├── dist │ └── localhost.yaml ├── exp │ ├── default.yaml │ └── fid_stats.yaml ├── ffhq256_uncond.yaml ├── fid.yaml ├── fid_stats.yaml ├── imagenet256_cond.yaml ├── imagenet256_uncond.yaml ├── imagenet512_cond.yaml ├── inception_score.yaml ├── loader │ ├── imagenet256_ddrm.yaml │ ├── imagenet256_ddrmpp.yaml │ ├── imagenet256_inception.yaml │ └── imagenet512_ddrmpp.yaml ├── model │ ├── ffhq256_uncond.yaml │ ├── imagenet256_cond.yaml │ ├── imagenet256_uncond.yaml │ └── imagenet512_cond.yaml ├── nonlinear_deblur_config.yaml ├── psnr.yaml └── stablediff │ └── params.yaml ├── algos ├── __init__.py ├── ddim.py ├── ddrm.py ├── deis.py ├── dps.py ├── identity.py ├── mcg.py ├── pgdm.py ├── reddiff.py ├── reddiff_parallel.py ├── sds.py └── sds_var.py ├── bkse ├── .gitignore ├── LICENSE ├── README.md ├── data │ ├── GOPRO_dataset.py │ ├── REDS_dataset.py │ ├── __init__.py │ ├── data_sampler.py │ ├── mix_dataset.py │ └── util.py ├── data_augmentation.py ├── domain_specific_deblur.py ├── experiments │ └── pretrained │ │ └── kernel.pth ├── generate_blur.py ├── generic_deblur.py ├── imgs │ ├── blur_faces │ │ └── face01.png │ ├── blur_imgs │ │ ├── blur1.png │ │ └── blur2.png │ ├── results │ │ ├── augmentation.jpg │ │ ├── domain_specific_deblur.jpg │ │ ├── general_deblurring.jpg │ │ ├── generate_blur.jpg │ │ └── kernel_encoding_wGT.png │ ├── sharp_imgs │ │ └── mushishi.png │ └── teaser.jpg ├── models │ ├── __init__.py │ ├── arch_util.py │ ├── backbones │ │ ├── resnet.py │ │ ├── skip │ │ │ ├── concat.py │ │ │ ├── downsampler.py │ │ │ ├── non_local_dot_product.py │ │ │ ├── skip.py │ │ │ └── util.py │ │ └── unet_parts.py │ ├── deblurring │ │ ├── image_deblur.py │ │ └── joint_deblur.py │ ├── dips.py │ ├── dsd │ │ ├── bicubic.py │ │ ├── dsd.py │ │ ├── dsd_stylegan.py │ │ ├── dsd_stylegan2.py │ │ ├── op │ │ │ ├── __init__.py │ │ │ ├── fused_act.py │ │ │ ├── fused_bias_act.cpp │ │ │ ├── fused_bias_act_kernel.cu │ │ │ ├── upfirdn2d.cpp │ │ │ ├── upfirdn2d.py │ │ │ └── upfirdn2d_kernel.cu │ │ ├── spherical_optimizer.py │ │ ├── stylegan.py │ │ └── stylegan2.py │ ├── kernel_encoding │ │ ├── base_model.py │ │ ├── image_base_model.py │ │ └── kernel_wizard.py │ ├── losses │ │ ├── charbonnier_loss.py │ │ ├── dsd_loss.py │ │ ├── gan_loss.py │ │ ├── hyper_laplacian_penalty.py │ │ ├── perceptual_loss.py │ │ └── ssim_loss.py │ └── lr_scheduler.py ├── options │ ├── __init__.py │ ├── data_augmentation │ │ └── default.yml │ ├── domain_specific_deblur │ │ ├── stylegan.yml │ │ └── stylegan2.yml │ ├── generate_blur │ │ └── default.yml │ ├── generic_deblur │ │ └── default.yml │ ├── kernel_encoding │ │ ├── GOPRO │ │ │ ├── wVAE.yml │ │ │ └── woVAE.yml │ │ ├── REDS │ │ │ └── woVAE.yml │ │ └── mix │ │ │ └── woVAE.yml │ └── options.py ├── requirements.txt ├── scripts │ ├── create_lmdb.py │ └── download_dataset.py ├── train.py ├── train_script.sh └── utils │ ├── __init__.py │ └── util.py ├── datasets ├── __init__.py ├── ffhq.py ├── imagenet.py ├── lmdb_dataset.py └── lsun.py ├── demo └── output.gif ├── eval ├── ca.py ├── fid.py ├── fid_stats.py ├── inception_score.py └── psnr.py ├── main.py ├── misc ├── dgp_top10.txt ├── dgp_top100.txt ├── dgp_top1k.txt ├── mcg_top1k.txt ├── palette_10k.txt ├── palette_5k.txt ├── palette_jpeg_demo.txt └── sr3_top1k.txt ├── models ├── __init__.py ├── classifier_guidance_model.py ├── diffusion.py └── guided_diffusion │ ├── __init__.py │ ├── fp16_util.py │ ├── logger.py │ ├── nn.py │ ├── script_util.py │ └── unet.py ├── motionblur ├── .gitignore ├── README.md ├── __init__.py ├── environment.yaml ├── example_kernel │ ├── kernel0.png │ ├── kernel100.png │ ├── kernel25.png │ ├── kernel50.png │ └── kernel75.png ├── images │ ├── flag.png │ ├── flagBLURRED.png │ └── moon.png ├── intensity.png └── motionblur.py ├── output.gif ├── playground ├── 100.png ├── 1000.png ├── 20.png ├── 50.png ├── 500.png ├── Untitled.ipynb ├── Untitled1.ipynb ├── adm.png ├── adm0.png ├── adm1.png ├── adm2.png ├── adm3.png ├── adm4.png ├── adm5.png ├── awd.png ├── awd_fwd.png ├── coltran_bot5k.txt ├── coltran_top5k.txt ├── compare_openai_sr_model.ipynb ├── ctest10k.txt ├── ddrmpp_res │ ├── 0.png │ ├── 10.png │ ├── 2.png │ ├── 20.png │ ├── 3.png │ └── 7.png ├── dgp_top1k.txt ├── figures.ipynb ├── jpeg5_deg │ ├── 0.png │ ├── 10.png │ ├── 2.png │ ├── 20.png │ ├── 3.png │ └── 7.png ├── jpeg5_ori │ ├── 0.png │ ├── 10.png │ ├── 2.png │ ├── 20.png │ ├── 3.png │ └── 7.png ├── lip.ipynb ├── palette_10k.txt ├── palette_img │ ├── 0.jpg │ ├── 10.jpg │ ├── 11.jpg │ ├── 2.jpg │ ├── 3.jpg │ └── 7.jpg ├── palette_result │ ├── 0.png │ ├── 10.png │ ├── 11.png │ ├── 2.png │ ├── 20.png │ ├── 3.png │ └── 7.png ├── plot_awd_steps.ipynb ├── process_imagenet_txt.ipynb └── svd.ipynb ├── requirements.txt ├── run_eval.sh ├── sample_batch.sh ├── sample_test.sh └── utils ├── __init__.py ├── checkpoints.py ├── dct.py ├── degredations.py ├── distributed.py ├── fft_utils.py ├── functions.py ├── jpeg_quantization.py ├── jpeg_torch.py └── save.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/README.md -------------------------------------------------------------------------------- /_configs/algo/ddim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/algo/ddim.yaml -------------------------------------------------------------------------------- /_configs/algo/ddrm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/algo/ddrm.yaml -------------------------------------------------------------------------------- /_configs/algo/dps.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/algo/dps.yaml -------------------------------------------------------------------------------- /_configs/algo/identity.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/algo/identity.yaml -------------------------------------------------------------------------------- /_configs/algo/mcg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/algo/mcg.yaml -------------------------------------------------------------------------------- /_configs/algo/pgdm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/algo/pgdm.yaml -------------------------------------------------------------------------------- /_configs/algo/reddiff.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/algo/reddiff.yaml -------------------------------------------------------------------------------- /_configs/algo/reddiff_parallel.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/algo/reddiff_parallel.yaml -------------------------------------------------------------------------------- /_configs/algo/sds.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/algo/sds.yaml -------------------------------------------------------------------------------- /_configs/algo/sds_var.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/algo/sds_var.yaml -------------------------------------------------------------------------------- /_configs/ca.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/ca.yaml -------------------------------------------------------------------------------- /_configs/classifier/imagenet256_cond.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/classifier/imagenet256_cond.yaml -------------------------------------------------------------------------------- /_configs/classifier/imagenet512_cond.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/classifier/imagenet512_cond.yaml -------------------------------------------------------------------------------- /_configs/classifier/none.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/classifier/none.yaml -------------------------------------------------------------------------------- /_configs/dataset/ffhq256_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/dataset/ffhq256_train.yaml -------------------------------------------------------------------------------- /_configs/dataset/ffhq256_val.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/dataset/ffhq256_val.yaml -------------------------------------------------------------------------------- /_configs/dataset/imagenet256_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/dataset/imagenet256_train.yaml -------------------------------------------------------------------------------- /_configs/dataset/imagenet256_val.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/dataset/imagenet256_val.yaml -------------------------------------------------------------------------------- /_configs/dataset/imagenet512_val.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/dataset/imagenet512_val.yaml -------------------------------------------------------------------------------- /_configs/ddrmpp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/ddrmpp.yaml -------------------------------------------------------------------------------- /_configs/ddrmpp_ffhq.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/ddrmpp_ffhq.yaml -------------------------------------------------------------------------------- /_configs/diffusion/linear1000.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/diffusion/linear1000.yaml -------------------------------------------------------------------------------- /_configs/dist/localhost.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/dist/localhost.yaml -------------------------------------------------------------------------------- /_configs/exp/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/exp/default.yaml -------------------------------------------------------------------------------- /_configs/exp/fid_stats.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/exp/fid_stats.yaml -------------------------------------------------------------------------------- /_configs/ffhq256_uncond.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/ffhq256_uncond.yaml -------------------------------------------------------------------------------- /_configs/fid.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/fid.yaml -------------------------------------------------------------------------------- /_configs/fid_stats.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/fid_stats.yaml -------------------------------------------------------------------------------- /_configs/imagenet256_cond.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/imagenet256_cond.yaml -------------------------------------------------------------------------------- /_configs/imagenet256_uncond.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/imagenet256_uncond.yaml -------------------------------------------------------------------------------- /_configs/imagenet512_cond.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/imagenet512_cond.yaml -------------------------------------------------------------------------------- /_configs/inception_score.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/inception_score.yaml -------------------------------------------------------------------------------- /_configs/loader/imagenet256_ddrm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/loader/imagenet256_ddrm.yaml -------------------------------------------------------------------------------- /_configs/loader/imagenet256_ddrmpp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/loader/imagenet256_ddrmpp.yaml -------------------------------------------------------------------------------- /_configs/loader/imagenet256_inception.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/loader/imagenet256_inception.yaml -------------------------------------------------------------------------------- /_configs/loader/imagenet512_ddrmpp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/loader/imagenet512_ddrmpp.yaml -------------------------------------------------------------------------------- /_configs/model/ffhq256_uncond.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/model/ffhq256_uncond.yaml -------------------------------------------------------------------------------- /_configs/model/imagenet256_cond.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/model/imagenet256_cond.yaml -------------------------------------------------------------------------------- /_configs/model/imagenet256_uncond.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/model/imagenet256_uncond.yaml -------------------------------------------------------------------------------- /_configs/model/imagenet512_cond.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/model/imagenet512_cond.yaml -------------------------------------------------------------------------------- /_configs/nonlinear_deblur_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/nonlinear_deblur_config.yaml -------------------------------------------------------------------------------- /_configs/psnr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/psnr.yaml -------------------------------------------------------------------------------- /_configs/stablediff/params.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/_configs/stablediff/params.yaml -------------------------------------------------------------------------------- /algos/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/algos/__init__.py -------------------------------------------------------------------------------- /algos/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/algos/ddim.py -------------------------------------------------------------------------------- /algos/ddrm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/algos/ddrm.py -------------------------------------------------------------------------------- /algos/deis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/algos/deis.py -------------------------------------------------------------------------------- /algos/dps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/algos/dps.py -------------------------------------------------------------------------------- /algos/identity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/algos/identity.py -------------------------------------------------------------------------------- /algos/mcg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/algos/mcg.py -------------------------------------------------------------------------------- /algos/pgdm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/algos/pgdm.py -------------------------------------------------------------------------------- /algos/reddiff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/algos/reddiff.py -------------------------------------------------------------------------------- /algos/reddiff_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/algos/reddiff_parallel.py -------------------------------------------------------------------------------- /algos/sds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/algos/sds.py -------------------------------------------------------------------------------- /algos/sds_var.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/algos/sds_var.py -------------------------------------------------------------------------------- /bkse/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/.gitignore -------------------------------------------------------------------------------- /bkse/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/LICENSE -------------------------------------------------------------------------------- /bkse/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/README.md -------------------------------------------------------------------------------- /bkse/data/GOPRO_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/data/GOPRO_dataset.py -------------------------------------------------------------------------------- /bkse/data/REDS_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/data/REDS_dataset.py -------------------------------------------------------------------------------- /bkse/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/data/__init__.py -------------------------------------------------------------------------------- /bkse/data/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/data/data_sampler.py -------------------------------------------------------------------------------- /bkse/data/mix_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/data/mix_dataset.py -------------------------------------------------------------------------------- /bkse/data/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/data/util.py -------------------------------------------------------------------------------- /bkse/data_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/data_augmentation.py -------------------------------------------------------------------------------- /bkse/domain_specific_deblur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/domain_specific_deblur.py -------------------------------------------------------------------------------- /bkse/experiments/pretrained/kernel.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/experiments/pretrained/kernel.pth -------------------------------------------------------------------------------- /bkse/generate_blur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/generate_blur.py -------------------------------------------------------------------------------- /bkse/generic_deblur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/generic_deblur.py -------------------------------------------------------------------------------- /bkse/imgs/blur_faces/face01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/imgs/blur_faces/face01.png -------------------------------------------------------------------------------- /bkse/imgs/blur_imgs/blur1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/imgs/blur_imgs/blur1.png -------------------------------------------------------------------------------- /bkse/imgs/blur_imgs/blur2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/imgs/blur_imgs/blur2.png -------------------------------------------------------------------------------- /bkse/imgs/results/augmentation.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/imgs/results/augmentation.jpg -------------------------------------------------------------------------------- /bkse/imgs/results/domain_specific_deblur.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/imgs/results/domain_specific_deblur.jpg -------------------------------------------------------------------------------- /bkse/imgs/results/general_deblurring.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/imgs/results/general_deblurring.jpg -------------------------------------------------------------------------------- /bkse/imgs/results/generate_blur.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/imgs/results/generate_blur.jpg -------------------------------------------------------------------------------- /bkse/imgs/results/kernel_encoding_wGT.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/imgs/results/kernel_encoding_wGT.png -------------------------------------------------------------------------------- /bkse/imgs/sharp_imgs/mushishi.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/imgs/sharp_imgs/mushishi.png -------------------------------------------------------------------------------- /bkse/imgs/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/imgs/teaser.jpg -------------------------------------------------------------------------------- /bkse/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/__init__.py -------------------------------------------------------------------------------- /bkse/models/arch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/arch_util.py -------------------------------------------------------------------------------- /bkse/models/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/backbones/resnet.py -------------------------------------------------------------------------------- /bkse/models/backbones/skip/concat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/backbones/skip/concat.py -------------------------------------------------------------------------------- /bkse/models/backbones/skip/downsampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/backbones/skip/downsampler.py -------------------------------------------------------------------------------- /bkse/models/backbones/skip/non_local_dot_product.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/backbones/skip/non_local_dot_product.py -------------------------------------------------------------------------------- /bkse/models/backbones/skip/skip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/backbones/skip/skip.py -------------------------------------------------------------------------------- /bkse/models/backbones/skip/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/backbones/skip/util.py -------------------------------------------------------------------------------- /bkse/models/backbones/unet_parts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/backbones/unet_parts.py -------------------------------------------------------------------------------- /bkse/models/deblurring/image_deblur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/deblurring/image_deblur.py -------------------------------------------------------------------------------- /bkse/models/deblurring/joint_deblur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/deblurring/joint_deblur.py -------------------------------------------------------------------------------- /bkse/models/dips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/dips.py -------------------------------------------------------------------------------- /bkse/models/dsd/bicubic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/dsd/bicubic.py -------------------------------------------------------------------------------- /bkse/models/dsd/dsd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/dsd/dsd.py -------------------------------------------------------------------------------- /bkse/models/dsd/dsd_stylegan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/dsd/dsd_stylegan.py -------------------------------------------------------------------------------- /bkse/models/dsd/dsd_stylegan2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/dsd/dsd_stylegan2.py -------------------------------------------------------------------------------- /bkse/models/dsd/op/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /bkse/models/dsd/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/dsd/op/fused_act.py -------------------------------------------------------------------------------- /bkse/models/dsd/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/dsd/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /bkse/models/dsd/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/dsd/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /bkse/models/dsd/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/dsd/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /bkse/models/dsd/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/dsd/op/upfirdn2d.py -------------------------------------------------------------------------------- /bkse/models/dsd/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/dsd/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /bkse/models/dsd/spherical_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/dsd/spherical_optimizer.py -------------------------------------------------------------------------------- /bkse/models/dsd/stylegan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/dsd/stylegan.py -------------------------------------------------------------------------------- /bkse/models/dsd/stylegan2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/dsd/stylegan2.py -------------------------------------------------------------------------------- /bkse/models/kernel_encoding/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/kernel_encoding/base_model.py -------------------------------------------------------------------------------- /bkse/models/kernel_encoding/image_base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/kernel_encoding/image_base_model.py -------------------------------------------------------------------------------- /bkse/models/kernel_encoding/kernel_wizard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/kernel_encoding/kernel_wizard.py -------------------------------------------------------------------------------- /bkse/models/losses/charbonnier_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/losses/charbonnier_loss.py -------------------------------------------------------------------------------- /bkse/models/losses/dsd_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/losses/dsd_loss.py -------------------------------------------------------------------------------- /bkse/models/losses/gan_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/losses/gan_loss.py -------------------------------------------------------------------------------- /bkse/models/losses/hyper_laplacian_penalty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/losses/hyper_laplacian_penalty.py -------------------------------------------------------------------------------- /bkse/models/losses/perceptual_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/losses/perceptual_loss.py -------------------------------------------------------------------------------- /bkse/models/losses/ssim_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/losses/ssim_loss.py -------------------------------------------------------------------------------- /bkse/models/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/models/lr_scheduler.py -------------------------------------------------------------------------------- /bkse/options/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /bkse/options/data_augmentation/default.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/options/data_augmentation/default.yml -------------------------------------------------------------------------------- /bkse/options/domain_specific_deblur/stylegan.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/options/domain_specific_deblur/stylegan.yml -------------------------------------------------------------------------------- /bkse/options/domain_specific_deblur/stylegan2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/options/domain_specific_deblur/stylegan2.yml -------------------------------------------------------------------------------- /bkse/options/generate_blur/default.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/options/generate_blur/default.yml -------------------------------------------------------------------------------- /bkse/options/generic_deblur/default.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/options/generic_deblur/default.yml -------------------------------------------------------------------------------- /bkse/options/kernel_encoding/GOPRO/wVAE.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/options/kernel_encoding/GOPRO/wVAE.yml -------------------------------------------------------------------------------- /bkse/options/kernel_encoding/GOPRO/woVAE.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/options/kernel_encoding/GOPRO/woVAE.yml -------------------------------------------------------------------------------- /bkse/options/kernel_encoding/REDS/woVAE.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/options/kernel_encoding/REDS/woVAE.yml -------------------------------------------------------------------------------- /bkse/options/kernel_encoding/mix/woVAE.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/options/kernel_encoding/mix/woVAE.yml -------------------------------------------------------------------------------- /bkse/options/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/options/options.py -------------------------------------------------------------------------------- /bkse/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/requirements.txt -------------------------------------------------------------------------------- /bkse/scripts/create_lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/scripts/create_lmdb.py -------------------------------------------------------------------------------- /bkse/scripts/download_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/scripts/download_dataset.py -------------------------------------------------------------------------------- /bkse/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/train.py -------------------------------------------------------------------------------- /bkse/train_script.sh: -------------------------------------------------------------------------------- 1 | python3.7 train.py -opt options/REDS/wsharp_woVAE.yml 2 | -------------------------------------------------------------------------------- /bkse/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /bkse/utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/bkse/utils/util.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/ffhq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/datasets/ffhq.py -------------------------------------------------------------------------------- /datasets/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/datasets/imagenet.py -------------------------------------------------------------------------------- /datasets/lmdb_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/datasets/lmdb_dataset.py -------------------------------------------------------------------------------- /datasets/lsun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/datasets/lsun.py -------------------------------------------------------------------------------- /demo/output.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/demo/output.gif -------------------------------------------------------------------------------- /eval/ca.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/eval/ca.py -------------------------------------------------------------------------------- /eval/fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/eval/fid.py -------------------------------------------------------------------------------- /eval/fid_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/eval/fid_stats.py -------------------------------------------------------------------------------- /eval/inception_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/eval/inception_score.py -------------------------------------------------------------------------------- /eval/psnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/eval/psnr.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/main.py -------------------------------------------------------------------------------- /misc/dgp_top10.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/misc/dgp_top10.txt -------------------------------------------------------------------------------- /misc/dgp_top100.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/misc/dgp_top100.txt -------------------------------------------------------------------------------- /misc/dgp_top1k.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/misc/dgp_top1k.txt -------------------------------------------------------------------------------- /misc/mcg_top1k.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/misc/mcg_top1k.txt -------------------------------------------------------------------------------- /misc/palette_10k.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/misc/palette_10k.txt -------------------------------------------------------------------------------- /misc/palette_5k.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/misc/palette_5k.txt -------------------------------------------------------------------------------- /misc/palette_jpeg_demo.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/misc/palette_jpeg_demo.txt -------------------------------------------------------------------------------- /misc/sr3_top1k.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/misc/sr3_top1k.txt -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/classifier_guidance_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/models/classifier_guidance_model.py -------------------------------------------------------------------------------- /models/diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/models/diffusion.py -------------------------------------------------------------------------------- /models/guided_diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/guided_diffusion/fp16_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/models/guided_diffusion/fp16_util.py -------------------------------------------------------------------------------- /models/guided_diffusion/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/models/guided_diffusion/logger.py -------------------------------------------------------------------------------- /models/guided_diffusion/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/models/guided_diffusion/nn.py -------------------------------------------------------------------------------- /models/guided_diffusion/script_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/models/guided_diffusion/script_util.py -------------------------------------------------------------------------------- /models/guided_diffusion/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/models/guided_diffusion/unet.py -------------------------------------------------------------------------------- /motionblur/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/motionblur/.gitignore -------------------------------------------------------------------------------- /motionblur/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/motionblur/README.md -------------------------------------------------------------------------------- /motionblur/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /motionblur/environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/motionblur/environment.yaml -------------------------------------------------------------------------------- /motionblur/example_kernel/kernel0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/motionblur/example_kernel/kernel0.png -------------------------------------------------------------------------------- /motionblur/example_kernel/kernel100.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/motionblur/example_kernel/kernel100.png -------------------------------------------------------------------------------- /motionblur/example_kernel/kernel25.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/motionblur/example_kernel/kernel25.png -------------------------------------------------------------------------------- /motionblur/example_kernel/kernel50.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/motionblur/example_kernel/kernel50.png -------------------------------------------------------------------------------- /motionblur/example_kernel/kernel75.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/motionblur/example_kernel/kernel75.png -------------------------------------------------------------------------------- /motionblur/images/flag.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/motionblur/images/flag.png -------------------------------------------------------------------------------- /motionblur/images/flagBLURRED.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/motionblur/images/flagBLURRED.png -------------------------------------------------------------------------------- /motionblur/images/moon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/motionblur/images/moon.png -------------------------------------------------------------------------------- /motionblur/intensity.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/motionblur/intensity.png -------------------------------------------------------------------------------- /motionblur/motionblur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/motionblur/motionblur.py -------------------------------------------------------------------------------- /output.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/output.gif -------------------------------------------------------------------------------- /playground/100.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/100.png -------------------------------------------------------------------------------- /playground/1000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/1000.png -------------------------------------------------------------------------------- /playground/20.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/20.png -------------------------------------------------------------------------------- /playground/50.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/50.png -------------------------------------------------------------------------------- /playground/500.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/500.png -------------------------------------------------------------------------------- /playground/Untitled.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/Untitled.ipynb -------------------------------------------------------------------------------- /playground/Untitled1.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/Untitled1.ipynb -------------------------------------------------------------------------------- /playground/adm.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/adm.png -------------------------------------------------------------------------------- /playground/adm0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/adm0.png -------------------------------------------------------------------------------- /playground/adm1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/adm1.png -------------------------------------------------------------------------------- /playground/adm2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/adm2.png -------------------------------------------------------------------------------- /playground/adm3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/adm3.png -------------------------------------------------------------------------------- /playground/adm4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/adm4.png -------------------------------------------------------------------------------- /playground/adm5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/adm5.png -------------------------------------------------------------------------------- /playground/awd.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/awd.png -------------------------------------------------------------------------------- /playground/awd_fwd.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/awd_fwd.png -------------------------------------------------------------------------------- /playground/coltran_bot5k.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/coltran_bot5k.txt -------------------------------------------------------------------------------- /playground/coltran_top5k.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/coltran_top5k.txt -------------------------------------------------------------------------------- /playground/compare_openai_sr_model.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/compare_openai_sr_model.ipynb -------------------------------------------------------------------------------- /playground/ctest10k.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/ctest10k.txt -------------------------------------------------------------------------------- /playground/ddrmpp_res/0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/ddrmpp_res/0.png -------------------------------------------------------------------------------- /playground/ddrmpp_res/10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/ddrmpp_res/10.png -------------------------------------------------------------------------------- /playground/ddrmpp_res/2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/ddrmpp_res/2.png -------------------------------------------------------------------------------- /playground/ddrmpp_res/20.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/ddrmpp_res/20.png -------------------------------------------------------------------------------- /playground/ddrmpp_res/3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/ddrmpp_res/3.png -------------------------------------------------------------------------------- /playground/ddrmpp_res/7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/ddrmpp_res/7.png -------------------------------------------------------------------------------- /playground/dgp_top1k.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/dgp_top1k.txt -------------------------------------------------------------------------------- /playground/figures.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/figures.ipynb -------------------------------------------------------------------------------- /playground/jpeg5_deg/0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/jpeg5_deg/0.png -------------------------------------------------------------------------------- /playground/jpeg5_deg/10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/jpeg5_deg/10.png -------------------------------------------------------------------------------- /playground/jpeg5_deg/2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/jpeg5_deg/2.png -------------------------------------------------------------------------------- /playground/jpeg5_deg/20.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/jpeg5_deg/20.png -------------------------------------------------------------------------------- /playground/jpeg5_deg/3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/jpeg5_deg/3.png -------------------------------------------------------------------------------- /playground/jpeg5_deg/7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/jpeg5_deg/7.png -------------------------------------------------------------------------------- /playground/jpeg5_ori/0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/jpeg5_ori/0.png -------------------------------------------------------------------------------- /playground/jpeg5_ori/10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/jpeg5_ori/10.png -------------------------------------------------------------------------------- /playground/jpeg5_ori/2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/jpeg5_ori/2.png -------------------------------------------------------------------------------- /playground/jpeg5_ori/20.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/jpeg5_ori/20.png -------------------------------------------------------------------------------- /playground/jpeg5_ori/3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/jpeg5_ori/3.png -------------------------------------------------------------------------------- /playground/jpeg5_ori/7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/jpeg5_ori/7.png -------------------------------------------------------------------------------- /playground/lip.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/lip.ipynb -------------------------------------------------------------------------------- /playground/palette_10k.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/palette_10k.txt -------------------------------------------------------------------------------- /playground/palette_img/0.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/palette_img/0.jpg -------------------------------------------------------------------------------- /playground/palette_img/10.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/palette_img/10.jpg -------------------------------------------------------------------------------- /playground/palette_img/11.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/palette_img/11.jpg -------------------------------------------------------------------------------- /playground/palette_img/2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/palette_img/2.jpg -------------------------------------------------------------------------------- /playground/palette_img/3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/palette_img/3.jpg -------------------------------------------------------------------------------- /playground/palette_img/7.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/palette_img/7.jpg -------------------------------------------------------------------------------- /playground/palette_result/0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/palette_result/0.png -------------------------------------------------------------------------------- /playground/palette_result/10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/palette_result/10.png -------------------------------------------------------------------------------- /playground/palette_result/11.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/palette_result/11.png -------------------------------------------------------------------------------- /playground/palette_result/2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/palette_result/2.png -------------------------------------------------------------------------------- /playground/palette_result/20.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/palette_result/20.png -------------------------------------------------------------------------------- /playground/palette_result/3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/palette_result/3.png -------------------------------------------------------------------------------- /playground/palette_result/7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/palette_result/7.png -------------------------------------------------------------------------------- /playground/plot_awd_steps.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/plot_awd_steps.ipynb -------------------------------------------------------------------------------- /playground/process_imagenet_txt.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/process_imagenet_txt.ipynb -------------------------------------------------------------------------------- /playground/svd.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/playground/svd.ipynb -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/run_eval.sh -------------------------------------------------------------------------------- /sample_batch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/sample_batch.sh -------------------------------------------------------------------------------- /sample_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/sample_test.sh -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/utils/checkpoints.py -------------------------------------------------------------------------------- /utils/dct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/utils/dct.py -------------------------------------------------------------------------------- /utils/degredations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/utils/degredations.py -------------------------------------------------------------------------------- /utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/utils/distributed.py -------------------------------------------------------------------------------- /utils/fft_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/utils/fft_utils.py -------------------------------------------------------------------------------- /utils/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/utils/functions.py -------------------------------------------------------------------------------- /utils/jpeg_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/utils/jpeg_quantization.py -------------------------------------------------------------------------------- /utils/jpeg_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/utils/jpeg_torch.py -------------------------------------------------------------------------------- /utils/save.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/RED-diff/HEAD/utils/save.py --------------------------------------------------------------------------------