├── .gitignore ├── README.md ├── basicsr ├── data │ ├── __init__.py │ ├── data_sampler.py │ ├── data_util.py │ ├── meta_info │ │ ├── meta_info_LHP_Rain_test.txt │ │ ├── meta_info_LHP_Rain_train.txt │ │ ├── meta_info_SPA_test.txt │ │ └── meta_info_SPA_train.txt │ ├── paired_image_dataset.py │ ├── prefetch_dataloader.py │ └── transforms.py ├── metrics │ ├── __init__.py │ ├── metric_util.py │ └── psnr_ssim.py ├── models │ ├── __init__.py │ ├── archs │ │ ├── DFSSM_arch.py │ │ ├── __init__.py │ │ └── basic_arch.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 ├── options ├── test_DFSSM_on_Rain200H.yml ├── test_DFSSM_on_SPA_Data.yml ├── train_DFSSM_on_Rain200H.yml └── train_DFSSM_on_SPA_Data.yml ├── requirements.txt ├── setup.cfg └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/README.md -------------------------------------------------------------------------------- /basicsr/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/data/__init__.py -------------------------------------------------------------------------------- /basicsr/data/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/data/data_sampler.py -------------------------------------------------------------------------------- /basicsr/data/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/data/data_util.py -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_LHP_Rain_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/data/meta_info/meta_info_LHP_Rain_test.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_LHP_Rain_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/data/meta_info/meta_info_LHP_Rain_train.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_SPA_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/data/meta_info/meta_info_SPA_test.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_SPA_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/data/meta_info/meta_info_SPA_train.txt -------------------------------------------------------------------------------- /basicsr/data/paired_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/data/paired_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/prefetch_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/data/prefetch_dataloader.py -------------------------------------------------------------------------------- /basicsr/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/data/transforms.py -------------------------------------------------------------------------------- /basicsr/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/metrics/__init__.py -------------------------------------------------------------------------------- /basicsr/metrics/metric_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/metrics/metric_util.py -------------------------------------------------------------------------------- /basicsr/metrics/psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/metrics/psnr_ssim.py -------------------------------------------------------------------------------- /basicsr/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/models/__init__.py -------------------------------------------------------------------------------- /basicsr/models/archs/DFSSM_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/models/archs/DFSSM_arch.py -------------------------------------------------------------------------------- /basicsr/models/archs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/models/archs/__init__.py -------------------------------------------------------------------------------- /basicsr/models/archs/basic_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/models/archs/basic_arch.py -------------------------------------------------------------------------------- /basicsr/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/models/base_model.py -------------------------------------------------------------------------------- /basicsr/models/image_restoration_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/models/image_restoration_model.py -------------------------------------------------------------------------------- /basicsr/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/models/losses/__init__.py -------------------------------------------------------------------------------- /basicsr/models/losses/loss_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/models/losses/loss_util.py -------------------------------------------------------------------------------- /basicsr/models/losses/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/models/losses/losses.py -------------------------------------------------------------------------------- /basicsr/models/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/models/lr_scheduler.py -------------------------------------------------------------------------------- /basicsr/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/test.py -------------------------------------------------------------------------------- /basicsr/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/train.py -------------------------------------------------------------------------------- /basicsr/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/utils/__init__.py -------------------------------------------------------------------------------- /basicsr/utils/bundle_submissions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/utils/bundle_submissions.py -------------------------------------------------------------------------------- /basicsr/utils/create_lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/utils/create_lmdb.py -------------------------------------------------------------------------------- /basicsr/utils/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/utils/dist_util.py -------------------------------------------------------------------------------- /basicsr/utils/download_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/utils/download_util.py -------------------------------------------------------------------------------- /basicsr/utils/face_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/utils/face_util.py -------------------------------------------------------------------------------- /basicsr/utils/file_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/utils/file_client.py -------------------------------------------------------------------------------- /basicsr/utils/flow_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/utils/flow_util.py -------------------------------------------------------------------------------- /basicsr/utils/img_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/utils/img_util.py -------------------------------------------------------------------------------- /basicsr/utils/lmdb_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/utils/lmdb_util.py -------------------------------------------------------------------------------- /basicsr/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/utils/logger.py -------------------------------------------------------------------------------- /basicsr/utils/matlab_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/utils/matlab_functions.py -------------------------------------------------------------------------------- /basicsr/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/utils/misc.py -------------------------------------------------------------------------------- /basicsr/utils/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/basicsr/utils/options.py -------------------------------------------------------------------------------- /options/test_DFSSM_on_Rain200H.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/options/test_DFSSM_on_Rain200H.yml -------------------------------------------------------------------------------- /options/test_DFSSM_on_SPA_Data.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/options/test_DFSSM_on_SPA_Data.yml -------------------------------------------------------------------------------- /options/train_DFSSM_on_Rain200H.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/options/train_DFSSM_on_Rain200H.yml -------------------------------------------------------------------------------- /options/train_DFSSM_on_SPA_Data.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/options/train_DFSSM_on_SPA_Data.yml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShugoYamashita/DFSSM/HEAD/setup.py --------------------------------------------------------------------------------