├── .vscode └── launch.json ├── README.md ├── VERSION ├── basicsr ├── __init__.py ├── archs │ ├── __init__.py │ ├── arch_util.py │ ├── gcfsr_arch.py │ └── vgg_arch.py ├── data │ ├── __init__.py │ ├── data_sampler.py │ ├── data_util.py │ ├── degradations.py │ ├── ffhq_control_dataset.py │ ├── meta_info │ │ ├── meta_info_DIV2K800sub_GT.txt │ │ ├── meta_info_REDS4_test_GT.txt │ │ ├── meta_info_REDS_GT.txt │ │ ├── meta_info_REDSofficial4_test_GT.txt │ │ ├── meta_info_REDSval_official_test_GT.txt │ │ ├── meta_info_Vimeo90K_test_GT.txt │ │ ├── meta_info_Vimeo90K_test_fast_GT.txt │ │ ├── meta_info_Vimeo90K_test_medium_GT.txt │ │ ├── meta_info_Vimeo90K_test_slow_GT.txt │ │ └── meta_info_Vimeo90K_train_GT.txt │ ├── paired_image_dataset.py │ ├── prefetch_dataloader.py │ ├── single_image_dataset.py │ ├── single_image_gt_dataset.py │ └── transforms.py ├── losses │ ├── __init__.py │ ├── loss_util.py │ └── losses.py ├── metrics │ ├── __init__.py │ ├── fid.py │ ├── metric_util.py │ ├── niqe.py │ ├── niqe_pris_params.npz │ └── psnr_ssim.py ├── models │ ├── __init__.py │ ├── base_model.py │ ├── gcfsr_model.py │ ├── lr_scheduler.py │ └── stylegan2_model.py ├── ops │ ├── __init__.py │ ├── fused_act │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-37.pyc │ │ │ ├── __init__.cpython-39.pyc │ │ │ ├── fused_act.cpython-37.pyc │ │ │ └── fused_act.cpython-39.pyc │ │ ├── fused_act.py │ │ ├── fused_act_ext.cpython-37m-x86_64-linux-gnu.so │ │ ├── fused_act_ext.cpython-39-x86_64-linux-gnu.so │ │ └── src │ │ │ ├── fused_bias_act.cpp │ │ │ └── fused_bias_act_kernel.cu │ └── upfirdn2d │ │ ├── __init__.py │ │ ├── __pycache__ │ │ ├── __init__.cpython-37.pyc │ │ ├── __init__.cpython-39.pyc │ │ ├── upfirdn2d.cpython-37.pyc │ │ └── upfirdn2d.cpython-39.pyc │ │ ├── src │ │ ├── upfirdn2d.cpp │ │ └── upfirdn2d_kernel.cu │ │ ├── upfirdn2d.py │ │ ├── upfirdn2d_ext.cpython-37m-x86_64-linux-gnu.so │ │ └── upfirdn2d_ext.cpython-39-x86_64-linux-gnu.so ├── test.py ├── train.py ├── utils │ ├── __init__.py │ ├── diffjpeg.py │ ├── dist_util.py │ ├── download_util.py │ ├── face_util.py │ ├── file_client.py │ ├── flow_util.py │ ├── img_process_util.py │ ├── img_util.py │ ├── lmdb_util.py │ ├── logger.py │ ├── matlab_functions.py │ ├── misc.py │ ├── options.py │ └── registry.py └── version.py ├── config ├── cat_liifsr3_x16.json ├── df2k_liifsr3_x4.json └── ffhq_liifsr3_scaler_16_128.json ├── core ├── logger.py ├── metrics.py └── wandb_logger.py ├── data ├── LRHR_dataset.py ├── __init__.py ├── prepare_data.py └── util.py ├── dist_util.py ├── idm_main.py ├── model ├── __init__.py ├── base_model.py ├── crop_validation.py ├── model.py ├── networks.py ├── ops │ ├── __init__.py │ ├── fused_act │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-37.pyc │ │ │ └── fused_act.cpython-37.pyc │ │ ├── fused_act.py │ │ ├── fused_act_ext.cpython-37m-x86_64-linux-gnu.so │ │ ├── fused_act_ext.cpython-39-x86_64-linux-gnu.so │ │ └── src │ │ │ ├── fused_bias_act.cpp │ │ │ └── fused_bias_act_kernel.cu │ └── upfirdn2d │ │ ├── __init__.py │ │ ├── __pycache__ │ │ ├── __init__.cpython-37.pyc │ │ └── upfirdn2d.cpython-37.pyc │ │ ├── src │ │ ├── upfirdn2d.cpp │ │ └── upfirdn2d_kernel.cu │ │ ├── upfirdn2d.py │ │ ├── upfirdn2d_ext.cpython-37m-x86_64-linux-gnu.so │ │ └── upfirdn2d_ext.cpython-39-x86_64-linux-gnu.so └── sr3_modules │ ├── diffusion.py │ ├── edsr.py │ ├── mlp.py │ ├── style.py │ └── unet.py ├── requirements.txt ├── run.sh ├── setup.cfg ├── setup.py └── utils.py /.vscode/launch.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/.vscode/launch.json -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/README.md -------------------------------------------------------------------------------- /VERSION: -------------------------------------------------------------------------------- 1 | 1.4.0 2 | -------------------------------------------------------------------------------- /basicsr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/__init__.py -------------------------------------------------------------------------------- /basicsr/archs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/archs/__init__.py -------------------------------------------------------------------------------- /basicsr/archs/arch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/archs/arch_util.py -------------------------------------------------------------------------------- /basicsr/archs/gcfsr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/archs/gcfsr_arch.py -------------------------------------------------------------------------------- /basicsr/archs/vgg_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/archs/vgg_arch.py -------------------------------------------------------------------------------- /basicsr/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/data/__init__.py -------------------------------------------------------------------------------- /basicsr/data/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/data/data_sampler.py -------------------------------------------------------------------------------- /basicsr/data/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/data/data_util.py -------------------------------------------------------------------------------- /basicsr/data/degradations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/data/degradations.py -------------------------------------------------------------------------------- /basicsr/data/ffhq_control_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/data/ffhq_control_dataset.py -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_DIV2K800sub_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/data/meta_info/meta_info_DIV2K800sub_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_REDS4_test_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/data/meta_info/meta_info_REDS4_test_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_REDS_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/data/meta_info/meta_info_REDS_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_REDSofficial4_test_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/data/meta_info/meta_info_REDSofficial4_test_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_REDSval_official_test_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/data/meta_info/meta_info_REDSval_official_test_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_Vimeo90K_test_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/data/meta_info/meta_info_Vimeo90K_test_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_Vimeo90K_test_fast_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/data/meta_info/meta_info_Vimeo90K_test_fast_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_Vimeo90K_test_medium_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/data/meta_info/meta_info_Vimeo90K_test_medium_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_Vimeo90K_test_slow_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/data/meta_info/meta_info_Vimeo90K_test_slow_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_Vimeo90K_train_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/data/meta_info/meta_info_Vimeo90K_train_GT.txt -------------------------------------------------------------------------------- /basicsr/data/paired_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/data/paired_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/prefetch_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/data/prefetch_dataloader.py -------------------------------------------------------------------------------- /basicsr/data/single_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/data/single_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/single_image_gt_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/data/single_image_gt_dataset.py -------------------------------------------------------------------------------- /basicsr/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/data/transforms.py -------------------------------------------------------------------------------- /basicsr/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/losses/__init__.py -------------------------------------------------------------------------------- /basicsr/losses/loss_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/losses/loss_util.py -------------------------------------------------------------------------------- /basicsr/losses/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/losses/losses.py -------------------------------------------------------------------------------- /basicsr/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/metrics/__init__.py -------------------------------------------------------------------------------- /basicsr/metrics/fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/metrics/fid.py -------------------------------------------------------------------------------- /basicsr/metrics/metric_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/metrics/metric_util.py -------------------------------------------------------------------------------- /basicsr/metrics/niqe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/metrics/niqe.py -------------------------------------------------------------------------------- /basicsr/metrics/niqe_pris_params.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/metrics/niqe_pris_params.npz -------------------------------------------------------------------------------- /basicsr/metrics/psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/metrics/psnr_ssim.py -------------------------------------------------------------------------------- /basicsr/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/models/__init__.py -------------------------------------------------------------------------------- /basicsr/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/models/base_model.py -------------------------------------------------------------------------------- /basicsr/models/gcfsr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/models/gcfsr_model.py -------------------------------------------------------------------------------- /basicsr/models/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/models/lr_scheduler.py -------------------------------------------------------------------------------- /basicsr/models/stylegan2_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/models/stylegan2_model.py -------------------------------------------------------------------------------- /basicsr/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /basicsr/ops/fused_act/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/ops/fused_act/__init__.py -------------------------------------------------------------------------------- /basicsr/ops/fused_act/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/ops/fused_act/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /basicsr/ops/fused_act/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/ops/fused_act/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/ops/fused_act/__pycache__/fused_act.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/ops/fused_act/__pycache__/fused_act.cpython-37.pyc -------------------------------------------------------------------------------- /basicsr/ops/fused_act/__pycache__/fused_act.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/ops/fused_act/__pycache__/fused_act.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/ops/fused_act/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/ops/fused_act/fused_act.py -------------------------------------------------------------------------------- /basicsr/ops/fused_act/fused_act_ext.cpython-37m-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/ops/fused_act/fused_act_ext.cpython-37m-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /basicsr/ops/fused_act/fused_act_ext.cpython-39-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/ops/fused_act/fused_act_ext.cpython-39-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /basicsr/ops/fused_act/src/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/ops/fused_act/src/fused_bias_act.cpp -------------------------------------------------------------------------------- /basicsr/ops/fused_act/src/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/ops/fused_act/src/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/ops/upfirdn2d/__init__.py -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/ops/upfirdn2d/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/ops/upfirdn2d/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/__pycache__/upfirdn2d.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/ops/upfirdn2d/__pycache__/upfirdn2d.cpython-37.pyc -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/__pycache__/upfirdn2d.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/ops/upfirdn2d/__pycache__/upfirdn2d.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/src/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/ops/upfirdn2d/src/upfirdn2d.cpp -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/src/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/ops/upfirdn2d/src/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/ops/upfirdn2d/upfirdn2d.py -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/upfirdn2d_ext.cpython-37m-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/ops/upfirdn2d/upfirdn2d_ext.cpython-37m-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /basicsr/ops/upfirdn2d/upfirdn2d_ext.cpython-39-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/ops/upfirdn2d/upfirdn2d_ext.cpython-39-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /basicsr/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/test.py -------------------------------------------------------------------------------- /basicsr/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/train.py -------------------------------------------------------------------------------- /basicsr/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/utils/__init__.py -------------------------------------------------------------------------------- /basicsr/utils/diffjpeg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/utils/diffjpeg.py -------------------------------------------------------------------------------- /basicsr/utils/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/utils/dist_util.py -------------------------------------------------------------------------------- /basicsr/utils/download_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/utils/download_util.py -------------------------------------------------------------------------------- /basicsr/utils/face_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/utils/face_util.py -------------------------------------------------------------------------------- /basicsr/utils/file_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/utils/file_client.py -------------------------------------------------------------------------------- /basicsr/utils/flow_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/utils/flow_util.py -------------------------------------------------------------------------------- /basicsr/utils/img_process_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/utils/img_process_util.py -------------------------------------------------------------------------------- /basicsr/utils/img_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/utils/img_util.py -------------------------------------------------------------------------------- /basicsr/utils/lmdb_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/utils/lmdb_util.py -------------------------------------------------------------------------------- /basicsr/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/utils/logger.py -------------------------------------------------------------------------------- /basicsr/utils/matlab_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/utils/matlab_functions.py -------------------------------------------------------------------------------- /basicsr/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/utils/misc.py -------------------------------------------------------------------------------- /basicsr/utils/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/utils/options.py -------------------------------------------------------------------------------- /basicsr/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/utils/registry.py -------------------------------------------------------------------------------- /basicsr/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/basicsr/version.py -------------------------------------------------------------------------------- /config/cat_liifsr3_x16.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/config/cat_liifsr3_x16.json -------------------------------------------------------------------------------- /config/df2k_liifsr3_x4.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/config/df2k_liifsr3_x4.json -------------------------------------------------------------------------------- /config/ffhq_liifsr3_scaler_16_128.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/config/ffhq_liifsr3_scaler_16_128.json -------------------------------------------------------------------------------- /core/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/core/logger.py -------------------------------------------------------------------------------- /core/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/core/metrics.py -------------------------------------------------------------------------------- /core/wandb_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/core/wandb_logger.py -------------------------------------------------------------------------------- /data/LRHR_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/data/LRHR_dataset.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/prepare_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/data/prepare_data.py -------------------------------------------------------------------------------- /data/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/data/util.py -------------------------------------------------------------------------------- /dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/dist_util.py -------------------------------------------------------------------------------- /idm_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/idm_main.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/base_model.py -------------------------------------------------------------------------------- /model/crop_validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/crop_validation.py -------------------------------------------------------------------------------- /model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/model.py -------------------------------------------------------------------------------- /model/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/networks.py -------------------------------------------------------------------------------- /model/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/ops/fused_act/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/ops/fused_act/__init__.py -------------------------------------------------------------------------------- /model/ops/fused_act/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/ops/fused_act/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /model/ops/fused_act/__pycache__/fused_act.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/ops/fused_act/__pycache__/fused_act.cpython-37.pyc -------------------------------------------------------------------------------- /model/ops/fused_act/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/ops/fused_act/fused_act.py -------------------------------------------------------------------------------- /model/ops/fused_act/fused_act_ext.cpython-37m-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/ops/fused_act/fused_act_ext.cpython-37m-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /model/ops/fused_act/fused_act_ext.cpython-39-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/ops/fused_act/fused_act_ext.cpython-39-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /model/ops/fused_act/src/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/ops/fused_act/src/fused_bias_act.cpp -------------------------------------------------------------------------------- /model/ops/fused_act/src/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/ops/fused_act/src/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /model/ops/upfirdn2d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/ops/upfirdn2d/__init__.py -------------------------------------------------------------------------------- /model/ops/upfirdn2d/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/ops/upfirdn2d/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /model/ops/upfirdn2d/__pycache__/upfirdn2d.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/ops/upfirdn2d/__pycache__/upfirdn2d.cpython-37.pyc -------------------------------------------------------------------------------- /model/ops/upfirdn2d/src/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/ops/upfirdn2d/src/upfirdn2d.cpp -------------------------------------------------------------------------------- /model/ops/upfirdn2d/src/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/ops/upfirdn2d/src/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /model/ops/upfirdn2d/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/ops/upfirdn2d/upfirdn2d.py -------------------------------------------------------------------------------- /model/ops/upfirdn2d/upfirdn2d_ext.cpython-37m-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/ops/upfirdn2d/upfirdn2d_ext.cpython-37m-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /model/ops/upfirdn2d/upfirdn2d_ext.cpython-39-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/ops/upfirdn2d/upfirdn2d_ext.cpython-39-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /model/sr3_modules/diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/sr3_modules/diffusion.py -------------------------------------------------------------------------------- /model/sr3_modules/edsr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/sr3_modules/edsr.py -------------------------------------------------------------------------------- /model/sr3_modules/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/sr3_modules/mlp.py -------------------------------------------------------------------------------- /model/sr3_modules/style.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/sr3_modules/style.py -------------------------------------------------------------------------------- /model/sr3_modules/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/model/sr3_modules/unet.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/run.sh -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/setup.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ree1s/IDM/HEAD/utils.py --------------------------------------------------------------------------------