├── Datasets └── README.md ├── Dehazing ├── Datasets │ └── README.md ├── Options │ ├── MB-TaylorFormer-B.yml │ └── MB-TaylorFormer-L.yml ├── README.md ├── __pycache__ │ └── utils.cpython-37.pyc ├── calculate_compare.py ├── download_data.py ├── evaluate_PSNR_SSIM.m ├── generate_AE.py ├── pretrained_models │ └── README.md ├── psnr_ ├── test.py └── utils.py ├── INSTALL.md ├── README.md ├── VERSION ├── basicsr ├── data │ ├── __init__.py │ ├── data_sampler.py │ ├── data_util.py │ ├── ffhq_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 │ ├── reds_dataset.py │ ├── single_image_dataset.py │ ├── transforms.py │ ├── video_test_dataset.py │ └── vimeo90k_dataset.py ├── metrics │ ├── __init__.py │ ├── fid.py │ ├── metric_util.py │ ├── niqe.py │ ├── niqe_pris_params.npz │ └── psnr_ssim.py ├── models │ ├── __init__.py │ ├── archs │ │ ├── MB_TaylorFormer.py │ │ ├── __init__.py │ │ └── arch_util.py │ ├── base_model.py │ ├── image_restoration_model.py │ ├── losses │ │ ├── __init__.py │ │ ├── loss_util.py │ │ └── losses.py │ └── lr_scheduler.py ├── test.py ├── train.py ├── utils │ ├── __init__.py │ ├── bundle_submissions.py │ ├── create_lmdb.py │ ├── dist_util.py │ ├── download_util.py │ ├── face_util.py │ ├── file_client.py │ ├── flow_util.py │ ├── img_util.py │ ├── lmdb_util.py │ ├── logger.py │ ├── matlab_functions.py │ ├── misc.py │ └── options.py └── version.py ├── figs └── pipeline.png ├── setup.cfg ├── setup.py ├── train.sh └── utils.py /Datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/Datasets/README.md -------------------------------------------------------------------------------- /Dehazing/Datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/Dehazing/Datasets/README.md -------------------------------------------------------------------------------- /Dehazing/Options/MB-TaylorFormer-B.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/Dehazing/Options/MB-TaylorFormer-B.yml -------------------------------------------------------------------------------- /Dehazing/Options/MB-TaylorFormer-L.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/Dehazing/Options/MB-TaylorFormer-L.yml -------------------------------------------------------------------------------- /Dehazing/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/Dehazing/README.md -------------------------------------------------------------------------------- /Dehazing/__pycache__/utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/Dehazing/__pycache__/utils.cpython-37.pyc -------------------------------------------------------------------------------- /Dehazing/calculate_compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/Dehazing/calculate_compare.py -------------------------------------------------------------------------------- /Dehazing/download_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/Dehazing/download_data.py -------------------------------------------------------------------------------- /Dehazing/evaluate_PSNR_SSIM.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/Dehazing/evaluate_PSNR_SSIM.m -------------------------------------------------------------------------------- /Dehazing/generate_AE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/Dehazing/generate_AE.py -------------------------------------------------------------------------------- /Dehazing/pretrained_models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/Dehazing/pretrained_models/README.md -------------------------------------------------------------------------------- /Dehazing/psnr_: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Dehazing/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/Dehazing/test.py -------------------------------------------------------------------------------- /Dehazing/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/Dehazing/utils.py -------------------------------------------------------------------------------- /INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/INSTALL.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/README.md -------------------------------------------------------------------------------- /VERSION: -------------------------------------------------------------------------------- 1 | 1.2.0 2 | -------------------------------------------------------------------------------- /basicsr/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/data/__init__.py -------------------------------------------------------------------------------- /basicsr/data/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/data/data_sampler.py -------------------------------------------------------------------------------- /basicsr/data/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/data/data_util.py -------------------------------------------------------------------------------- /basicsr/data/ffhq_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/data/ffhq_dataset.py -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_DIV2K800sub_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/data/meta_info/meta_info_DIV2K800sub_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_REDS4_test_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/data/meta_info/meta_info_REDS4_test_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_REDS_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/data/meta_info/meta_info_REDS_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_REDSofficial4_test_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/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/FVL2020/ICCV-2023-MB-TaylorFormer/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/FVL2020/ICCV-2023-MB-TaylorFormer/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/FVL2020/ICCV-2023-MB-TaylorFormer/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/FVL2020/ICCV-2023-MB-TaylorFormer/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/FVL2020/ICCV-2023-MB-TaylorFormer/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/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/data/meta_info/meta_info_Vimeo90K_train_GT.txt -------------------------------------------------------------------------------- /basicsr/data/paired_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/data/paired_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/prefetch_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/data/prefetch_dataloader.py -------------------------------------------------------------------------------- /basicsr/data/reds_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/data/reds_dataset.py -------------------------------------------------------------------------------- /basicsr/data/single_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/data/single_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/data/transforms.py -------------------------------------------------------------------------------- /basicsr/data/video_test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/data/video_test_dataset.py -------------------------------------------------------------------------------- /basicsr/data/vimeo90k_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/data/vimeo90k_dataset.py -------------------------------------------------------------------------------- /basicsr/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/metrics/__init__.py -------------------------------------------------------------------------------- /basicsr/metrics/fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/metrics/fid.py -------------------------------------------------------------------------------- /basicsr/metrics/metric_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/metrics/metric_util.py -------------------------------------------------------------------------------- /basicsr/metrics/niqe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/metrics/niqe.py -------------------------------------------------------------------------------- /basicsr/metrics/niqe_pris_params.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/metrics/niqe_pris_params.npz -------------------------------------------------------------------------------- /basicsr/metrics/psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/metrics/psnr_ssim.py -------------------------------------------------------------------------------- /basicsr/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/models/__init__.py -------------------------------------------------------------------------------- /basicsr/models/archs/MB_TaylorFormer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/models/archs/MB_TaylorFormer.py -------------------------------------------------------------------------------- /basicsr/models/archs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/models/archs/__init__.py -------------------------------------------------------------------------------- /basicsr/models/archs/arch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/models/archs/arch_util.py -------------------------------------------------------------------------------- /basicsr/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/models/base_model.py -------------------------------------------------------------------------------- /basicsr/models/image_restoration_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/models/image_restoration_model.py -------------------------------------------------------------------------------- /basicsr/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/models/losses/__init__.py -------------------------------------------------------------------------------- /basicsr/models/losses/loss_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/models/losses/loss_util.py -------------------------------------------------------------------------------- /basicsr/models/losses/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/models/losses/losses.py -------------------------------------------------------------------------------- /basicsr/models/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/models/lr_scheduler.py -------------------------------------------------------------------------------- /basicsr/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/test.py -------------------------------------------------------------------------------- /basicsr/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/train.py -------------------------------------------------------------------------------- /basicsr/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/utils/__init__.py -------------------------------------------------------------------------------- /basicsr/utils/bundle_submissions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/utils/bundle_submissions.py -------------------------------------------------------------------------------- /basicsr/utils/create_lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/utils/create_lmdb.py -------------------------------------------------------------------------------- /basicsr/utils/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/utils/dist_util.py -------------------------------------------------------------------------------- /basicsr/utils/download_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/utils/download_util.py -------------------------------------------------------------------------------- /basicsr/utils/face_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/utils/face_util.py -------------------------------------------------------------------------------- /basicsr/utils/file_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/utils/file_client.py -------------------------------------------------------------------------------- /basicsr/utils/flow_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/utils/flow_util.py -------------------------------------------------------------------------------- /basicsr/utils/img_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/utils/img_util.py -------------------------------------------------------------------------------- /basicsr/utils/lmdb_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/utils/lmdb_util.py -------------------------------------------------------------------------------- /basicsr/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/utils/logger.py -------------------------------------------------------------------------------- /basicsr/utils/matlab_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/utils/matlab_functions.py -------------------------------------------------------------------------------- /basicsr/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/utils/misc.py -------------------------------------------------------------------------------- /basicsr/utils/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/utils/options.py -------------------------------------------------------------------------------- /basicsr/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/basicsr/version.py -------------------------------------------------------------------------------- /figs/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/figs/pipeline.png -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/setup.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/train.sh -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FVL2020/ICCV-2023-MB-TaylorFormer/HEAD/utils.py --------------------------------------------------------------------------------