├── .gitignore ├── .vscode └── launch.json ├── LICENSE ├── README.md ├── _configs ├── algo │ ├── ddim.yaml │ ├── ddrm.yaml │ ├── deft.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 ├── deft.yaml ├── diffusion │ └── linear1000.yaml ├── dist │ └── localhost.yaml ├── eval_imagenet256_uncond.yaml ├── exp │ ├── default.yaml │ └── fid_stats.yaml ├── ffhq256_uncond.yaml ├── fid.yaml ├── fid_stats.yaml ├── finetune_dataset │ └── imagenet256_finetune.yaml ├── htransform_model │ └── imagenet256_htransform.yaml ├── imagenet256_cond.yaml ├── imagenet256_deft.yaml ├── imagenet256_uncond.yaml ├── imagenet512_cond.yaml ├── inception_score.yaml ├── likelihood │ ├── blur.yaml │ ├── ct.yaml │ ├── hdr.yaml │ ├── inp.yaml │ ├── phase_retrieval.yaml │ └── sr4.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 └── wandb_config │ └── shreyaspadhy.yaml ├── algos ├── __init__.py ├── ddim.py ├── ddrm.py ├── deft.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 ├── aapm.py ├── data │ ├── dgp_top1k.txt │ ├── imagenet256_train_mean_std.npz │ ├── imagenet_10k.txt │ ├── imagenet_1k_val.txt │ └── meta.bin ├── ffhq.py ├── imagenet.py ├── lmdb_dataset.py ├── lodopab.py └── lsun.py ├── eval.py ├── eval ├── __init__.py ├── ca.py ├── fid.py ├── fid_stats.py ├── inception_score.py └── psnr.py ├── htransform ├── likelihoods.py ├── losses.py └── trainer.py ├── images └── DEFT_Sampling.png ├── 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 │ └── unet_cond.py └── utils.py ├── pretrained_model └── Readme.md ├── pyproject.toml ├── radon ├── functional.py └── tomography.py ├── run_commands.md ├── run_eval.sh ├── sample_batch.sh ├── sample_test.sh ├── test.png ├── utils ├── __init__.py ├── checkpoints.py ├── dct.py ├── degredations.py ├── distributed.py ├── fft_utils.py ├── functions.py ├── jpeg_quantization.py ├── jpeg_torch.py ├── preprocessing.py └── save.py └── uv.lock /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/.gitignore -------------------------------------------------------------------------------- /.vscode/launch.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/.vscode/launch.json -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/README.md -------------------------------------------------------------------------------- /_configs/algo/ddim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/algo/ddim.yaml -------------------------------------------------------------------------------- /_configs/algo/ddrm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/algo/ddrm.yaml -------------------------------------------------------------------------------- /_configs/algo/deft.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/algo/deft.yaml -------------------------------------------------------------------------------- /_configs/algo/dps.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/algo/dps.yaml -------------------------------------------------------------------------------- /_configs/algo/identity.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/algo/identity.yaml -------------------------------------------------------------------------------- /_configs/algo/mcg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/algo/mcg.yaml -------------------------------------------------------------------------------- /_configs/algo/pgdm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/algo/pgdm.yaml -------------------------------------------------------------------------------- /_configs/algo/reddiff.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/algo/reddiff.yaml -------------------------------------------------------------------------------- /_configs/algo/reddiff_parallel.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/algo/reddiff_parallel.yaml -------------------------------------------------------------------------------- /_configs/algo/sds.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/algo/sds.yaml -------------------------------------------------------------------------------- /_configs/algo/sds_var.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/algo/sds_var.yaml -------------------------------------------------------------------------------- /_configs/ca.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/ca.yaml -------------------------------------------------------------------------------- /_configs/classifier/imagenet256_cond.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/classifier/imagenet256_cond.yaml -------------------------------------------------------------------------------- /_configs/classifier/imagenet512_cond.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/classifier/imagenet512_cond.yaml -------------------------------------------------------------------------------- /_configs/classifier/none.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/classifier/none.yaml -------------------------------------------------------------------------------- /_configs/dataset/ffhq256_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/dataset/ffhq256_train.yaml -------------------------------------------------------------------------------- /_configs/dataset/ffhq256_val.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/dataset/ffhq256_val.yaml -------------------------------------------------------------------------------- /_configs/dataset/imagenet256_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/dataset/imagenet256_train.yaml -------------------------------------------------------------------------------- /_configs/dataset/imagenet256_val.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/dataset/imagenet256_val.yaml -------------------------------------------------------------------------------- /_configs/dataset/imagenet512_val.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/dataset/imagenet512_val.yaml -------------------------------------------------------------------------------- /_configs/ddrmpp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/ddrmpp.yaml -------------------------------------------------------------------------------- /_configs/ddrmpp_ffhq.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/ddrmpp_ffhq.yaml -------------------------------------------------------------------------------- /_configs/deft.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/deft.yaml -------------------------------------------------------------------------------- /_configs/diffusion/linear1000.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/diffusion/linear1000.yaml -------------------------------------------------------------------------------- /_configs/dist/localhost.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/dist/localhost.yaml -------------------------------------------------------------------------------- /_configs/eval_imagenet256_uncond.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/eval_imagenet256_uncond.yaml -------------------------------------------------------------------------------- /_configs/exp/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/exp/default.yaml -------------------------------------------------------------------------------- /_configs/exp/fid_stats.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/exp/fid_stats.yaml -------------------------------------------------------------------------------- /_configs/ffhq256_uncond.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/ffhq256_uncond.yaml -------------------------------------------------------------------------------- /_configs/fid.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/fid.yaml -------------------------------------------------------------------------------- /_configs/fid_stats.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/fid_stats.yaml -------------------------------------------------------------------------------- /_configs/finetune_dataset/imagenet256_finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/finetune_dataset/imagenet256_finetune.yaml -------------------------------------------------------------------------------- /_configs/htransform_model/imagenet256_htransform.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/htransform_model/imagenet256_htransform.yaml -------------------------------------------------------------------------------- /_configs/imagenet256_cond.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/imagenet256_cond.yaml -------------------------------------------------------------------------------- /_configs/imagenet256_deft.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/imagenet256_deft.yaml -------------------------------------------------------------------------------- /_configs/imagenet256_uncond.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/imagenet256_uncond.yaml -------------------------------------------------------------------------------- /_configs/imagenet512_cond.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/imagenet512_cond.yaml -------------------------------------------------------------------------------- /_configs/inception_score.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/inception_score.yaml -------------------------------------------------------------------------------- /_configs/likelihood/blur.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/likelihood/blur.yaml -------------------------------------------------------------------------------- /_configs/likelihood/ct.yaml: -------------------------------------------------------------------------------- 1 | name: ct 2 | 3 | forward_op: 4 | num_angles: 60 5 | noise_std: 0.1 6 | -------------------------------------------------------------------------------- /_configs/likelihood/hdr.yaml: -------------------------------------------------------------------------------- 1 | name: hdr 2 | 3 | forward_op: 4 | noise_std: 0.1 5 | -------------------------------------------------------------------------------- /_configs/likelihood/inp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/likelihood/inp.yaml -------------------------------------------------------------------------------- /_configs/likelihood/phase_retrieval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/likelihood/phase_retrieval.yaml -------------------------------------------------------------------------------- /_configs/likelihood/sr4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/likelihood/sr4.yaml -------------------------------------------------------------------------------- /_configs/loader/imagenet256_ddrm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/loader/imagenet256_ddrm.yaml -------------------------------------------------------------------------------- /_configs/loader/imagenet256_ddrmpp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/loader/imagenet256_ddrmpp.yaml -------------------------------------------------------------------------------- /_configs/loader/imagenet256_inception.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/loader/imagenet256_inception.yaml -------------------------------------------------------------------------------- /_configs/loader/imagenet512_ddrmpp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/loader/imagenet512_ddrmpp.yaml -------------------------------------------------------------------------------- /_configs/model/ffhq256_uncond.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/model/ffhq256_uncond.yaml -------------------------------------------------------------------------------- /_configs/model/imagenet256_cond.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/model/imagenet256_cond.yaml -------------------------------------------------------------------------------- /_configs/model/imagenet256_uncond.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/model/imagenet256_uncond.yaml -------------------------------------------------------------------------------- /_configs/model/imagenet512_cond.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/model/imagenet512_cond.yaml -------------------------------------------------------------------------------- /_configs/nonlinear_deblur_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/nonlinear_deblur_config.yaml -------------------------------------------------------------------------------- /_configs/psnr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/psnr.yaml -------------------------------------------------------------------------------- /_configs/stablediff/params.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/stablediff/params.yaml -------------------------------------------------------------------------------- /_configs/wandb_config/shreyaspadhy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/_configs/wandb_config/shreyaspadhy.yaml -------------------------------------------------------------------------------- /algos/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/algos/__init__.py -------------------------------------------------------------------------------- /algos/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/algos/ddim.py -------------------------------------------------------------------------------- /algos/ddrm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/algos/ddrm.py -------------------------------------------------------------------------------- /algos/deft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/algos/deft.py -------------------------------------------------------------------------------- /algos/deis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/algos/deis.py -------------------------------------------------------------------------------- /algos/dps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/algos/dps.py -------------------------------------------------------------------------------- /algos/identity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/algos/identity.py -------------------------------------------------------------------------------- /algos/mcg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/algos/mcg.py -------------------------------------------------------------------------------- /algos/pgdm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/algos/pgdm.py -------------------------------------------------------------------------------- /algos/reddiff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/algos/reddiff.py -------------------------------------------------------------------------------- /algos/reddiff_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/algos/reddiff_parallel.py -------------------------------------------------------------------------------- /algos/sds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/algos/sds.py -------------------------------------------------------------------------------- /algos/sds_var.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/algos/sds_var.py -------------------------------------------------------------------------------- /bkse/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/.gitignore -------------------------------------------------------------------------------- /bkse/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/LICENSE -------------------------------------------------------------------------------- /bkse/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/README.md -------------------------------------------------------------------------------- /bkse/data/GOPRO_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/data/GOPRO_dataset.py -------------------------------------------------------------------------------- /bkse/data/REDS_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/data/REDS_dataset.py -------------------------------------------------------------------------------- /bkse/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/data/__init__.py -------------------------------------------------------------------------------- /bkse/data/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/data/data_sampler.py -------------------------------------------------------------------------------- /bkse/data/mix_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/data/mix_dataset.py -------------------------------------------------------------------------------- /bkse/data/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/data/util.py -------------------------------------------------------------------------------- /bkse/data_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/data_augmentation.py -------------------------------------------------------------------------------- /bkse/domain_specific_deblur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/domain_specific_deblur.py -------------------------------------------------------------------------------- /bkse/experiments/pretrained/kernel.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/experiments/pretrained/kernel.pth -------------------------------------------------------------------------------- /bkse/generate_blur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/generate_blur.py -------------------------------------------------------------------------------- /bkse/generic_deblur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/generic_deblur.py -------------------------------------------------------------------------------- /bkse/imgs/blur_faces/face01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/imgs/blur_faces/face01.png -------------------------------------------------------------------------------- /bkse/imgs/blur_imgs/blur1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/imgs/blur_imgs/blur1.png -------------------------------------------------------------------------------- /bkse/imgs/blur_imgs/blur2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/imgs/blur_imgs/blur2.png -------------------------------------------------------------------------------- /bkse/imgs/results/augmentation.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/imgs/results/augmentation.jpg -------------------------------------------------------------------------------- /bkse/imgs/results/domain_specific_deblur.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/imgs/results/domain_specific_deblur.jpg -------------------------------------------------------------------------------- /bkse/imgs/results/general_deblurring.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/imgs/results/general_deblurring.jpg -------------------------------------------------------------------------------- /bkse/imgs/results/generate_blur.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/imgs/results/generate_blur.jpg -------------------------------------------------------------------------------- /bkse/imgs/results/kernel_encoding_wGT.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/imgs/results/kernel_encoding_wGT.png -------------------------------------------------------------------------------- /bkse/imgs/sharp_imgs/mushishi.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/imgs/sharp_imgs/mushishi.png -------------------------------------------------------------------------------- /bkse/imgs/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/imgs/teaser.jpg -------------------------------------------------------------------------------- /bkse/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/__init__.py -------------------------------------------------------------------------------- /bkse/models/arch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/arch_util.py -------------------------------------------------------------------------------- /bkse/models/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/backbones/resnet.py -------------------------------------------------------------------------------- /bkse/models/backbones/skip/concat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/backbones/skip/concat.py -------------------------------------------------------------------------------- /bkse/models/backbones/skip/downsampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/backbones/skip/downsampler.py -------------------------------------------------------------------------------- /bkse/models/backbones/skip/non_local_dot_product.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/backbones/skip/non_local_dot_product.py -------------------------------------------------------------------------------- /bkse/models/backbones/skip/skip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/backbones/skip/skip.py -------------------------------------------------------------------------------- /bkse/models/backbones/skip/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/backbones/skip/util.py -------------------------------------------------------------------------------- /bkse/models/backbones/unet_parts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/backbones/unet_parts.py -------------------------------------------------------------------------------- /bkse/models/deblurring/image_deblur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/deblurring/image_deblur.py -------------------------------------------------------------------------------- /bkse/models/deblurring/joint_deblur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/deblurring/joint_deblur.py -------------------------------------------------------------------------------- /bkse/models/dips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/dips.py -------------------------------------------------------------------------------- /bkse/models/dsd/bicubic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/dsd/bicubic.py -------------------------------------------------------------------------------- /bkse/models/dsd/dsd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/dsd/dsd.py -------------------------------------------------------------------------------- /bkse/models/dsd/dsd_stylegan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/dsd/dsd_stylegan.py -------------------------------------------------------------------------------- /bkse/models/dsd/dsd_stylegan2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/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/alexdenker/DEFT/HEAD/bkse/models/dsd/op/fused_act.py -------------------------------------------------------------------------------- /bkse/models/dsd/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/dsd/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /bkse/models/dsd/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/dsd/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /bkse/models/dsd/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/dsd/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /bkse/models/dsd/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/dsd/op/upfirdn2d.py -------------------------------------------------------------------------------- /bkse/models/dsd/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/dsd/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /bkse/models/dsd/spherical_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/dsd/spherical_optimizer.py -------------------------------------------------------------------------------- /bkse/models/dsd/stylegan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/dsd/stylegan.py -------------------------------------------------------------------------------- /bkse/models/dsd/stylegan2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/dsd/stylegan2.py -------------------------------------------------------------------------------- /bkse/models/kernel_encoding/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/kernel_encoding/base_model.py -------------------------------------------------------------------------------- /bkse/models/kernel_encoding/image_base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/kernel_encoding/image_base_model.py -------------------------------------------------------------------------------- /bkse/models/kernel_encoding/kernel_wizard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/kernel_encoding/kernel_wizard.py -------------------------------------------------------------------------------- /bkse/models/losses/charbonnier_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/losses/charbonnier_loss.py -------------------------------------------------------------------------------- /bkse/models/losses/dsd_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/losses/dsd_loss.py -------------------------------------------------------------------------------- /bkse/models/losses/gan_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/losses/gan_loss.py -------------------------------------------------------------------------------- /bkse/models/losses/hyper_laplacian_penalty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/losses/hyper_laplacian_penalty.py -------------------------------------------------------------------------------- /bkse/models/losses/perceptual_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/losses/perceptual_loss.py -------------------------------------------------------------------------------- /bkse/models/losses/ssim_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/losses/ssim_loss.py -------------------------------------------------------------------------------- /bkse/models/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/models/lr_scheduler.py -------------------------------------------------------------------------------- /bkse/options/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /bkse/options/data_augmentation/default.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/options/data_augmentation/default.yml -------------------------------------------------------------------------------- /bkse/options/domain_specific_deblur/stylegan.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/options/domain_specific_deblur/stylegan.yml -------------------------------------------------------------------------------- /bkse/options/domain_specific_deblur/stylegan2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/options/domain_specific_deblur/stylegan2.yml -------------------------------------------------------------------------------- /bkse/options/generate_blur/default.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/options/generate_blur/default.yml -------------------------------------------------------------------------------- /bkse/options/generic_deblur/default.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/options/generic_deblur/default.yml -------------------------------------------------------------------------------- /bkse/options/kernel_encoding/GOPRO/wVAE.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/options/kernel_encoding/GOPRO/wVAE.yml -------------------------------------------------------------------------------- /bkse/options/kernel_encoding/GOPRO/woVAE.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/options/kernel_encoding/GOPRO/woVAE.yml -------------------------------------------------------------------------------- /bkse/options/kernel_encoding/REDS/woVAE.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/options/kernel_encoding/REDS/woVAE.yml -------------------------------------------------------------------------------- /bkse/options/kernel_encoding/mix/woVAE.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/options/kernel_encoding/mix/woVAE.yml -------------------------------------------------------------------------------- /bkse/options/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/options/options.py -------------------------------------------------------------------------------- /bkse/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/requirements.txt -------------------------------------------------------------------------------- /bkse/scripts/create_lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/scripts/create_lmdb.py -------------------------------------------------------------------------------- /bkse/scripts/download_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/bkse/scripts/download_dataset.py -------------------------------------------------------------------------------- /bkse/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/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/alexdenker/DEFT/HEAD/bkse/utils/util.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/aapm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/datasets/aapm.py -------------------------------------------------------------------------------- /datasets/data/dgp_top1k.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/datasets/data/dgp_top1k.txt -------------------------------------------------------------------------------- /datasets/data/imagenet256_train_mean_std.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/datasets/data/imagenet256_train_mean_std.npz -------------------------------------------------------------------------------- /datasets/data/imagenet_10k.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/datasets/data/imagenet_10k.txt -------------------------------------------------------------------------------- /datasets/data/imagenet_1k_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/datasets/data/imagenet_1k_val.txt -------------------------------------------------------------------------------- /datasets/data/meta.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/datasets/data/meta.bin -------------------------------------------------------------------------------- /datasets/ffhq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/datasets/ffhq.py -------------------------------------------------------------------------------- /datasets/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/datasets/imagenet.py -------------------------------------------------------------------------------- /datasets/lmdb_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/datasets/lmdb_dataset.py -------------------------------------------------------------------------------- /datasets/lodopab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/datasets/lodopab.py -------------------------------------------------------------------------------- /datasets/lsun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/datasets/lsun.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/eval.py -------------------------------------------------------------------------------- /eval/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /eval/ca.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/eval/ca.py -------------------------------------------------------------------------------- /eval/fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/eval/fid.py -------------------------------------------------------------------------------- /eval/fid_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/eval/fid_stats.py -------------------------------------------------------------------------------- /eval/inception_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/eval/inception_score.py -------------------------------------------------------------------------------- /eval/psnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/eval/psnr.py -------------------------------------------------------------------------------- /htransform/likelihoods.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/htransform/likelihoods.py -------------------------------------------------------------------------------- /htransform/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/htransform/losses.py -------------------------------------------------------------------------------- /htransform/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/htransform/trainer.py -------------------------------------------------------------------------------- /images/DEFT_Sampling.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/images/DEFT_Sampling.png -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/main.py -------------------------------------------------------------------------------- /misc/dgp_top10.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/misc/dgp_top10.txt -------------------------------------------------------------------------------- /misc/dgp_top100.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/misc/dgp_top100.txt -------------------------------------------------------------------------------- /misc/dgp_top1k.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/misc/dgp_top1k.txt -------------------------------------------------------------------------------- /misc/mcg_top1k.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/misc/mcg_top1k.txt -------------------------------------------------------------------------------- /misc/palette_10k.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/misc/palette_10k.txt -------------------------------------------------------------------------------- /misc/palette_5k.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/misc/palette_5k.txt -------------------------------------------------------------------------------- /misc/palette_jpeg_demo.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/misc/palette_jpeg_demo.txt -------------------------------------------------------------------------------- /misc/sr3_top1k.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/misc/sr3_top1k.txt -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/classifier_guidance_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/models/classifier_guidance_model.py -------------------------------------------------------------------------------- /models/diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/models/diffusion.py -------------------------------------------------------------------------------- /models/guided_diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/guided_diffusion/fp16_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/models/guided_diffusion/fp16_util.py -------------------------------------------------------------------------------- /models/guided_diffusion/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/models/guided_diffusion/logger.py -------------------------------------------------------------------------------- /models/guided_diffusion/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/models/guided_diffusion/nn.py -------------------------------------------------------------------------------- /models/guided_diffusion/script_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/models/guided_diffusion/script_util.py -------------------------------------------------------------------------------- /models/guided_diffusion/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/models/guided_diffusion/unet.py -------------------------------------------------------------------------------- /models/guided_diffusion/unet_cond.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/models/guided_diffusion/unet_cond.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/models/utils.py -------------------------------------------------------------------------------- /pretrained_model/Readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/pretrained_model/Readme.md -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/pyproject.toml -------------------------------------------------------------------------------- /radon/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/radon/functional.py -------------------------------------------------------------------------------- /radon/tomography.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/radon/tomography.py -------------------------------------------------------------------------------- /run_commands.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/run_commands.md -------------------------------------------------------------------------------- /run_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/run_eval.sh -------------------------------------------------------------------------------- /sample_batch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/sample_batch.sh -------------------------------------------------------------------------------- /sample_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/sample_test.sh -------------------------------------------------------------------------------- /test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/test.png -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/utils/checkpoints.py -------------------------------------------------------------------------------- /utils/dct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/utils/dct.py -------------------------------------------------------------------------------- /utils/degredations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/utils/degredations.py -------------------------------------------------------------------------------- /utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/utils/distributed.py -------------------------------------------------------------------------------- /utils/fft_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/utils/fft_utils.py -------------------------------------------------------------------------------- /utils/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/utils/functions.py -------------------------------------------------------------------------------- /utils/jpeg_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/utils/jpeg_quantization.py -------------------------------------------------------------------------------- /utils/jpeg_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/utils/jpeg_torch.py -------------------------------------------------------------------------------- /utils/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/utils/preprocessing.py -------------------------------------------------------------------------------- /utils/save.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/utils/save.py -------------------------------------------------------------------------------- /uv.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alexdenker/DEFT/HEAD/uv.lock --------------------------------------------------------------------------------