├── Figures └── DRCNet.png ├── LICENSE ├── README.md ├── basicsr ├── data │ ├── __init__.py │ ├── data_sampler.py │ ├── data_util.py │ ├── feature.py │ ├── hist.py │ ├── paired_image_dataset.py │ ├── prefetch_dataloader.py │ ├── single_image_dataset.py │ └── transforms.py ├── metrics │ ├── __init__.py │ ├── metric_util.py │ └── psnr_ssim.py ├── models │ ├── __init__.py │ ├── archs │ │ ├── __init__.py │ │ ├── arch_util.py │ │ ├── dcon.py │ │ ├── drcnet_arch.py │ │ └── norm.py │ ├── base_model.py │ ├── image_restoration_model.py │ ├── losses │ │ ├── __init__.py │ │ ├── loss_util.py │ │ └── losses.py │ └── lr_scheduler.py ├── test.py ├── train.py ├── train_rain.py └── utils │ ├── __init__.py │ ├── create_lmdb.py │ ├── dist_util.py │ ├── download_util.py │ ├── file_client.py │ ├── img_util.py │ ├── lmdb_util.py │ ├── logger.py │ ├── matlab_functions.py │ ├── misc.py │ └── options.py ├── options ├── test │ ├── GoPro │ │ └── DRCNet-GoPro.yml │ ├── Rain13k │ │ ├── DRCNet-Rain100H.yml │ │ ├── DRCNet-Rain100L.yml │ │ ├── DRCNet-Test100.yml │ │ ├── DRCNet-Test1200.yml │ │ └── DRCNet-Test2800.yml │ └── SIDD │ │ └── DRCNet.yml └── train │ ├── GoPro │ └── DRCNet.yml │ ├── Rain13k │ └── DRCNet.yml │ └── SIDD │ └── DRCNet.yml ├── requirements.txt ├── scripts ├── data_preparation │ ├── gopro.py │ ├── rain13k.py │ └── sidd.py ├── download_gdrive.py ├── download_pretrained_models.py └── publish_models.py ├── setup.cfg └── setup.py /Figures/DRCNet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/Figures/DRCNet.png -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/README.md -------------------------------------------------------------------------------- /basicsr/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/data/__init__.py -------------------------------------------------------------------------------- /basicsr/data/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/data/data_sampler.py -------------------------------------------------------------------------------- /basicsr/data/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/data/data_util.py -------------------------------------------------------------------------------- /basicsr/data/feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/data/feature.py -------------------------------------------------------------------------------- /basicsr/data/hist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/data/hist.py -------------------------------------------------------------------------------- /basicsr/data/paired_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/data/paired_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/prefetch_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/data/prefetch_dataloader.py -------------------------------------------------------------------------------- /basicsr/data/single_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/data/single_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/data/transforms.py -------------------------------------------------------------------------------- /basicsr/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/metrics/__init__.py -------------------------------------------------------------------------------- /basicsr/metrics/metric_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/metrics/metric_util.py -------------------------------------------------------------------------------- /basicsr/metrics/psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/metrics/psnr_ssim.py -------------------------------------------------------------------------------- /basicsr/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/models/__init__.py -------------------------------------------------------------------------------- /basicsr/models/archs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/models/archs/__init__.py -------------------------------------------------------------------------------- /basicsr/models/archs/arch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/models/archs/arch_util.py -------------------------------------------------------------------------------- /basicsr/models/archs/dcon.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/models/archs/dcon.py -------------------------------------------------------------------------------- /basicsr/models/archs/drcnet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/models/archs/drcnet_arch.py -------------------------------------------------------------------------------- /basicsr/models/archs/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/models/archs/norm.py -------------------------------------------------------------------------------- /basicsr/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/models/base_model.py -------------------------------------------------------------------------------- /basicsr/models/image_restoration_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/models/image_restoration_model.py -------------------------------------------------------------------------------- /basicsr/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/models/losses/__init__.py -------------------------------------------------------------------------------- /basicsr/models/losses/loss_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/models/losses/loss_util.py -------------------------------------------------------------------------------- /basicsr/models/losses/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/models/losses/losses.py -------------------------------------------------------------------------------- /basicsr/models/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/models/lr_scheduler.py -------------------------------------------------------------------------------- /basicsr/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/test.py -------------------------------------------------------------------------------- /basicsr/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/train.py -------------------------------------------------------------------------------- /basicsr/train_rain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/train_rain.py -------------------------------------------------------------------------------- /basicsr/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/utils/__init__.py -------------------------------------------------------------------------------- /basicsr/utils/create_lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/utils/create_lmdb.py -------------------------------------------------------------------------------- /basicsr/utils/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/utils/dist_util.py -------------------------------------------------------------------------------- /basicsr/utils/download_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/utils/download_util.py -------------------------------------------------------------------------------- /basicsr/utils/file_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/utils/file_client.py -------------------------------------------------------------------------------- /basicsr/utils/img_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/utils/img_util.py -------------------------------------------------------------------------------- /basicsr/utils/lmdb_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/utils/lmdb_util.py -------------------------------------------------------------------------------- /basicsr/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/utils/logger.py -------------------------------------------------------------------------------- /basicsr/utils/matlab_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/utils/matlab_functions.py -------------------------------------------------------------------------------- /basicsr/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/utils/misc.py -------------------------------------------------------------------------------- /basicsr/utils/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/basicsr/utils/options.py -------------------------------------------------------------------------------- /options/test/GoPro/DRCNet-GoPro.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/options/test/GoPro/DRCNet-GoPro.yml -------------------------------------------------------------------------------- /options/test/Rain13k/DRCNet-Rain100H.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/options/test/Rain13k/DRCNet-Rain100H.yml -------------------------------------------------------------------------------- /options/test/Rain13k/DRCNet-Rain100L.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/options/test/Rain13k/DRCNet-Rain100L.yml -------------------------------------------------------------------------------- /options/test/Rain13k/DRCNet-Test100.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/options/test/Rain13k/DRCNet-Test100.yml -------------------------------------------------------------------------------- /options/test/Rain13k/DRCNet-Test1200.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/options/test/Rain13k/DRCNet-Test1200.yml -------------------------------------------------------------------------------- /options/test/Rain13k/DRCNet-Test2800.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/options/test/Rain13k/DRCNet-Test2800.yml -------------------------------------------------------------------------------- /options/test/SIDD/DRCNet.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/options/test/SIDD/DRCNet.yml -------------------------------------------------------------------------------- /options/train/GoPro/DRCNet.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/options/train/GoPro/DRCNet.yml -------------------------------------------------------------------------------- /options/train/Rain13k/DRCNet.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/options/train/Rain13k/DRCNet.yml -------------------------------------------------------------------------------- /options/train/SIDD/DRCNet.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/options/train/SIDD/DRCNet.yml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/data_preparation/gopro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/scripts/data_preparation/gopro.py -------------------------------------------------------------------------------- /scripts/data_preparation/rain13k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/scripts/data_preparation/rain13k.py -------------------------------------------------------------------------------- /scripts/data_preparation/sidd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/scripts/data_preparation/sidd.py -------------------------------------------------------------------------------- /scripts/download_gdrive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/scripts/download_gdrive.py -------------------------------------------------------------------------------- /scripts/download_pretrained_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/scripts/download_pretrained_models.py -------------------------------------------------------------------------------- /scripts/publish_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/scripts/publish_models.py -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leefly072/DRCNet/HEAD/setup.py --------------------------------------------------------------------------------