├── .gitignore ├── LICENSE ├── README.md ├── configs ├── config_base.yaml ├── df2k4x.yaml ├── diffsr_base.yaml ├── diffsr_highresnet.yaml ├── diffsr_hr_ltae.yaml ├── diffsr_no_encoder.yaml ├── diffsr_rrdb_ltae.yaml ├── diffsr_sat.yaml ├── misr │ ├── highresnet.yaml │ ├── median_misr_pretrain.yaml │ ├── misr_pretrain.yaml │ ├── rrdb_highresnet.yaml │ └── rrdb_ltae.yaml ├── rrdb │ ├── rrdb.yaml │ └── rrdb_8_blocks.yaml └── sr_base.yaml ├── data_gen ├── dataset.py ├── hist.py ├── minmax.py └── preprocess_data.py ├── models ├── HRNet.py ├── commons.py ├── config_hrnet.json ├── config_rrdb_misr.json ├── diffsr_modules.py ├── diffusion.py ├── misr_module.py ├── module_util.py └── positional_encoding.py ├── requirements.txt ├── tasks ├── bicubic.py ├── config_hrnet.json ├── config_rrdb_misr.json ├── highresnet.py ├── hr_ltae.py ├── rrdb.py ├── rrdb_highresnet_fusion.py ├── rrdb_ltae.py ├── srdiff.py ├── srdiff_sat.py └── trainer.py └── utils ├── dataloader.py ├── dataloader_utils.py ├── hparams.py ├── matlab_resize.py ├── sr_utils.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | .ipynb_checkpoints/ 2 | .venv/ 3 | __pycache__/ 4 | 5 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/README.md -------------------------------------------------------------------------------- /configs/config_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/configs/config_base.yaml -------------------------------------------------------------------------------- /configs/df2k4x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/configs/df2k4x.yaml -------------------------------------------------------------------------------- /configs/diffsr_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/configs/diffsr_base.yaml -------------------------------------------------------------------------------- /configs/diffsr_highresnet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/configs/diffsr_highresnet.yaml -------------------------------------------------------------------------------- /configs/diffsr_hr_ltae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/configs/diffsr_hr_ltae.yaml -------------------------------------------------------------------------------- /configs/diffsr_no_encoder.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/configs/diffsr_no_encoder.yaml -------------------------------------------------------------------------------- /configs/diffsr_rrdb_ltae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/configs/diffsr_rrdb_ltae.yaml -------------------------------------------------------------------------------- /configs/diffsr_sat.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/configs/diffsr_sat.yaml -------------------------------------------------------------------------------- /configs/misr/highresnet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/configs/misr/highresnet.yaml -------------------------------------------------------------------------------- /configs/misr/median_misr_pretrain.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/configs/misr/median_misr_pretrain.yaml -------------------------------------------------------------------------------- /configs/misr/misr_pretrain.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/configs/misr/misr_pretrain.yaml -------------------------------------------------------------------------------- /configs/misr/rrdb_highresnet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/configs/misr/rrdb_highresnet.yaml -------------------------------------------------------------------------------- /configs/misr/rrdb_ltae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/configs/misr/rrdb_ltae.yaml -------------------------------------------------------------------------------- /configs/rrdb/rrdb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/configs/rrdb/rrdb.yaml -------------------------------------------------------------------------------- /configs/rrdb/rrdb_8_blocks.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/configs/rrdb/rrdb_8_blocks.yaml -------------------------------------------------------------------------------- /configs/sr_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/configs/sr_base.yaml -------------------------------------------------------------------------------- /data_gen/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/data_gen/dataset.py -------------------------------------------------------------------------------- /data_gen/hist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/data_gen/hist.py -------------------------------------------------------------------------------- /data_gen/minmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/data_gen/minmax.py -------------------------------------------------------------------------------- /data_gen/preprocess_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/data_gen/preprocess_data.py -------------------------------------------------------------------------------- /models/HRNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/models/HRNet.py -------------------------------------------------------------------------------- /models/commons.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/models/commons.py -------------------------------------------------------------------------------- /models/config_hrnet.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/models/config_hrnet.json -------------------------------------------------------------------------------- /models/config_rrdb_misr.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/models/config_rrdb_misr.json -------------------------------------------------------------------------------- /models/diffsr_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/models/diffsr_modules.py -------------------------------------------------------------------------------- /models/diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/models/diffusion.py -------------------------------------------------------------------------------- /models/misr_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/models/misr_module.py -------------------------------------------------------------------------------- /models/module_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/models/module_util.py -------------------------------------------------------------------------------- /models/positional_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/models/positional_encoding.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/requirements.txt -------------------------------------------------------------------------------- /tasks/bicubic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/tasks/bicubic.py -------------------------------------------------------------------------------- /tasks/config_hrnet.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/tasks/config_hrnet.json -------------------------------------------------------------------------------- /tasks/config_rrdb_misr.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/tasks/config_rrdb_misr.json -------------------------------------------------------------------------------- /tasks/highresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/tasks/highresnet.py -------------------------------------------------------------------------------- /tasks/hr_ltae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/tasks/hr_ltae.py -------------------------------------------------------------------------------- /tasks/rrdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/tasks/rrdb.py -------------------------------------------------------------------------------- /tasks/rrdb_highresnet_fusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/tasks/rrdb_highresnet_fusion.py -------------------------------------------------------------------------------- /tasks/rrdb_ltae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/tasks/rrdb_ltae.py -------------------------------------------------------------------------------- /tasks/srdiff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/tasks/srdiff.py -------------------------------------------------------------------------------- /tasks/srdiff_sat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/tasks/srdiff_sat.py -------------------------------------------------------------------------------- /tasks/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/tasks/trainer.py -------------------------------------------------------------------------------- /utils/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/utils/dataloader.py -------------------------------------------------------------------------------- /utils/dataloader_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/utils/dataloader_utils.py -------------------------------------------------------------------------------- /utils/hparams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/utils/hparams.py -------------------------------------------------------------------------------- /utils/matlab_resize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/utils/matlab_resize.py -------------------------------------------------------------------------------- /utils/sr_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/utils/sr_utils.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimiokab/MISR-S2/HEAD/utils/utils.py --------------------------------------------------------------------------------