├── .gitignore ├── CREDITS.md ├── LICENSE ├── README.md ├── configs ├── brn │ ├── README.md │ └── brn_c32r8d3d5.py ├── cgan │ ├── README.md │ └── cgan_gc64dc48.py ├── dcsfn │ ├── README.md │ ├── dcsfn_c20s4l16_gru.py │ ├── dcsfn_c20s4l16_lstm.py │ └── dcsfn_c20s4l16_rnn.py ├── ddn │ ├── README.md │ ├── ddn_c16d26.py │ └── ddn_c16d26_rain1400.py ├── derainnet │ ├── README.md │ └── derainnet_c512.py ├── drdnet │ ├── README.md │ └── drdnet_c64d16.py ├── dual_gcn │ ├── README.md │ └── dual_gcn_c72n11.py ├── ecnet │ ├── README.md │ ├── ecnet_stage1_c32d4.py │ ├── ecnet_stage2_c32d4.py │ └── ecnet_stage2_c32d4r6_ll.py ├── lpnet │ ├── README.md │ └── lpnet_c16d5r5.py ├── mardnet │ ├── README.md │ └── mard_c32g32n8.py ├── oucdnet │ ├── README.md │ └── oucdnet_o3n5_msff.py ├── physical_model_guided │ ├── README.md │ └── pmg_c64d5s3.py ├── prenet │ ├── README.md │ ├── prenet_c32s6d5_gru.py │ ├── prenet_c32s6d5_gru_r.py │ ├── prenet_c32s6d5_lstm.py │ ├── prenet_c32s6d5_lstm_r.py │ ├── prn_c32s6d5.py │ └── prn_c32s6d5_r.py ├── rcdnet │ ├── README.md │ └── rcdnet_c32s17n4.py ├── rehen │ ├── README.md │ └── rehen_c24s4n5.py ├── rescan │ ├── README.md │ ├── rescan_c24s4d5_gru_add.py │ ├── rescan_c24s4d5_gru_full.py │ ├── rescan_c24s4d5_lstm_add.py │ ├── rescan_c24s4d5_lstm_full.py │ ├── rescan_c24s4d5_rnn_add.py │ └── rescan_c24s4d5_rnn_full.py ├── rlnet │ ├── README.md │ ├── rlnet_official_code.py │ ├── rlnet_stage1.py │ └── rlnet_stage2.py └── spdnet │ ├── README.md │ └── spdnet_c32s3_l3ns3nr3.py ├── dist_test.sh ├── dist_train.sh ├── docs ├── dataset_preparation.md ├── getting_started.md └── installation.md ├── figs ├── brn.png ├── dcsfn.png ├── ddn.png ├── derainnet.png ├── drdnet.png ├── dual_gcn.png ├── ecnet.png ├── id_cgan.png ├── lpnet.png ├── mardnet.png ├── oucdnet.png ├── physical_model_guided.png ├── prenet.png ├── rcdnet.png ├── rehen.png ├── rescan.png ├── rlnet.png └── spdnet.png ├── get_flops.py ├── get_model_info.py ├── mmderain ├── __init__.py ├── apis │ ├── __init__.py │ ├── test.py │ └── train.py ├── core │ ├── __init__.py │ ├── distributed_wrapper.py │ ├── evaluation │ │ ├── __init__.py │ │ ├── eval_hooks.py │ │ ├── metric_utils.py │ │ ├── metrics.py │ │ └── niqe_pris_params.npz │ ├── hooks │ │ ├── __init__.py │ │ └── rlnet_hyperparam_hook.py │ ├── initializers │ │ ├── __init__.py │ │ └── ecnet_initializer.py │ ├── misc.py │ ├── optimizer │ │ ├── __init__.py │ │ └── builder.py │ └── scheduler │ │ ├── __init__.py │ │ └── lr_updater.py ├── datasets │ ├── __init__.py │ ├── base_dataset.py │ ├── base_derain_dataset.py │ ├── builder.py │ ├── dataset_wrappers.py │ ├── derain_filename_matching_dataset.py │ ├── derain_paired_dataset.py │ ├── derain_unpaired_dataset.py │ ├── pipelines │ │ ├── __init__.py │ │ ├── augmentation.py │ │ ├── compose.py │ │ ├── crop.py │ │ ├── formating.py │ │ ├── loading.py │ │ ├── normalization.py │ │ └── utils.py │ ├── registry.py │ └── samplers │ │ ├── __init__.py │ │ └── distributed_sampler.py ├── models │ ├── __init__.py │ ├── backbones │ │ ├── __init__.py │ │ ├── brn.py │ │ ├── cgan.py │ │ ├── dcsfn.py │ │ ├── ddn.py │ │ ├── derainnet.py │ │ ├── drdnet.py │ │ ├── dual_gcn.py │ │ ├── ecnet.py │ │ ├── lpnet.py │ │ ├── mardnet.py │ │ ├── oucdnet.py │ │ ├── physical_model_guided.py │ │ ├── prenet.py │ │ ├── rcdnet.py │ │ ├── rehen.py │ │ ├── rescan.py │ │ ├── rlnet.py │ │ └── spdnet.py │ ├── base.py │ ├── builder.py │ ├── common │ │ ├── __init__.py │ │ ├── box_filter.py │ │ ├── guided_filter.py │ │ ├── model_utils.py │ │ ├── priors.py │ │ ├── pyramids.py │ │ └── utils.py │ ├── layers │ │ ├── __init__.py │ │ ├── ecnet.py │ │ ├── mprnet.py │ │ ├── rescan.py │ │ ├── rnn.py │ │ └── se_module.py │ ├── losses │ │ ├── __init__.py │ │ ├── gan_loss.py │ │ ├── gradient_loss.py │ │ ├── perceptual_loss.py │ │ ├── pixelwise_loss.py │ │ ├── structural_loss.py │ │ └── utils.py │ ├── registry.py │ └── restorers │ │ ├── __init__.py │ │ ├── basic_restorer.py │ │ ├── cgan.py │ │ ├── ecnet.py │ │ ├── lpnet.py │ │ ├── multi_output_restorer.py │ │ ├── multi_stage_restorer.py │ │ ├── physical_model_guided.py │ │ ├── rcdnet.py │ │ ├── rlnet.py │ │ └── spdnet.py ├── utils │ ├── __init__.py │ ├── functools.py │ └── logger.py └── version.py ├── pylintrc ├── requirements.txt ├── requirements ├── runtime.txt └── tests.txt ├── test.py ├── tools └── publish_model.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/.gitignore -------------------------------------------------------------------------------- /CREDITS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/CREDITS.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/README.md -------------------------------------------------------------------------------- /configs/brn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/brn/README.md -------------------------------------------------------------------------------- /configs/brn/brn_c32r8d3d5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/brn/brn_c32r8d3d5.py -------------------------------------------------------------------------------- /configs/cgan/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/cgan/README.md -------------------------------------------------------------------------------- /configs/cgan/cgan_gc64dc48.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/cgan/cgan_gc64dc48.py -------------------------------------------------------------------------------- /configs/dcsfn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/dcsfn/README.md -------------------------------------------------------------------------------- /configs/dcsfn/dcsfn_c20s4l16_gru.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/dcsfn/dcsfn_c20s4l16_gru.py -------------------------------------------------------------------------------- /configs/dcsfn/dcsfn_c20s4l16_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/dcsfn/dcsfn_c20s4l16_lstm.py -------------------------------------------------------------------------------- /configs/dcsfn/dcsfn_c20s4l16_rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/dcsfn/dcsfn_c20s4l16_rnn.py -------------------------------------------------------------------------------- /configs/ddn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/ddn/README.md -------------------------------------------------------------------------------- /configs/ddn/ddn_c16d26.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/ddn/ddn_c16d26.py -------------------------------------------------------------------------------- /configs/ddn/ddn_c16d26_rain1400.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/ddn/ddn_c16d26_rain1400.py -------------------------------------------------------------------------------- /configs/derainnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/derainnet/README.md -------------------------------------------------------------------------------- /configs/derainnet/derainnet_c512.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/derainnet/derainnet_c512.py -------------------------------------------------------------------------------- /configs/drdnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/drdnet/README.md -------------------------------------------------------------------------------- /configs/drdnet/drdnet_c64d16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/drdnet/drdnet_c64d16.py -------------------------------------------------------------------------------- /configs/dual_gcn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/dual_gcn/README.md -------------------------------------------------------------------------------- /configs/dual_gcn/dual_gcn_c72n11.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/dual_gcn/dual_gcn_c72n11.py -------------------------------------------------------------------------------- /configs/ecnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/ecnet/README.md -------------------------------------------------------------------------------- /configs/ecnet/ecnet_stage1_c32d4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/ecnet/ecnet_stage1_c32d4.py -------------------------------------------------------------------------------- /configs/ecnet/ecnet_stage2_c32d4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/ecnet/ecnet_stage2_c32d4.py -------------------------------------------------------------------------------- /configs/ecnet/ecnet_stage2_c32d4r6_ll.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/ecnet/ecnet_stage2_c32d4r6_ll.py -------------------------------------------------------------------------------- /configs/lpnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/lpnet/README.md -------------------------------------------------------------------------------- /configs/lpnet/lpnet_c16d5r5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/lpnet/lpnet_c16d5r5.py -------------------------------------------------------------------------------- /configs/mardnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/mardnet/README.md -------------------------------------------------------------------------------- /configs/mardnet/mard_c32g32n8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/mardnet/mard_c32g32n8.py -------------------------------------------------------------------------------- /configs/oucdnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/oucdnet/README.md -------------------------------------------------------------------------------- /configs/oucdnet/oucdnet_o3n5_msff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/oucdnet/oucdnet_o3n5_msff.py -------------------------------------------------------------------------------- /configs/physical_model_guided/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/physical_model_guided/README.md -------------------------------------------------------------------------------- /configs/physical_model_guided/pmg_c64d5s3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/physical_model_guided/pmg_c64d5s3.py -------------------------------------------------------------------------------- /configs/prenet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/prenet/README.md -------------------------------------------------------------------------------- /configs/prenet/prenet_c32s6d5_gru.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/prenet/prenet_c32s6d5_gru.py -------------------------------------------------------------------------------- /configs/prenet/prenet_c32s6d5_gru_r.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/prenet/prenet_c32s6d5_gru_r.py -------------------------------------------------------------------------------- /configs/prenet/prenet_c32s6d5_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/prenet/prenet_c32s6d5_lstm.py -------------------------------------------------------------------------------- /configs/prenet/prenet_c32s6d5_lstm_r.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/prenet/prenet_c32s6d5_lstm_r.py -------------------------------------------------------------------------------- /configs/prenet/prn_c32s6d5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/prenet/prn_c32s6d5.py -------------------------------------------------------------------------------- /configs/prenet/prn_c32s6d5_r.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/prenet/prn_c32s6d5_r.py -------------------------------------------------------------------------------- /configs/rcdnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/rcdnet/README.md -------------------------------------------------------------------------------- /configs/rcdnet/rcdnet_c32s17n4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/rcdnet/rcdnet_c32s17n4.py -------------------------------------------------------------------------------- /configs/rehen/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/rehen/README.md -------------------------------------------------------------------------------- /configs/rehen/rehen_c24s4n5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/rehen/rehen_c24s4n5.py -------------------------------------------------------------------------------- /configs/rescan/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/rescan/README.md -------------------------------------------------------------------------------- /configs/rescan/rescan_c24s4d5_gru_add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/rescan/rescan_c24s4d5_gru_add.py -------------------------------------------------------------------------------- /configs/rescan/rescan_c24s4d5_gru_full.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/rescan/rescan_c24s4d5_gru_full.py -------------------------------------------------------------------------------- /configs/rescan/rescan_c24s4d5_lstm_add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/rescan/rescan_c24s4d5_lstm_add.py -------------------------------------------------------------------------------- /configs/rescan/rescan_c24s4d5_lstm_full.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/rescan/rescan_c24s4d5_lstm_full.py -------------------------------------------------------------------------------- /configs/rescan/rescan_c24s4d5_rnn_add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/rescan/rescan_c24s4d5_rnn_add.py -------------------------------------------------------------------------------- /configs/rescan/rescan_c24s4d5_rnn_full.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/rescan/rescan_c24s4d5_rnn_full.py -------------------------------------------------------------------------------- /configs/rlnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/rlnet/README.md -------------------------------------------------------------------------------- /configs/rlnet/rlnet_official_code.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/rlnet/rlnet_official_code.py -------------------------------------------------------------------------------- /configs/rlnet/rlnet_stage1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/rlnet/rlnet_stage1.py -------------------------------------------------------------------------------- /configs/rlnet/rlnet_stage2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/rlnet/rlnet_stage2.py -------------------------------------------------------------------------------- /configs/spdnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/spdnet/README.md -------------------------------------------------------------------------------- /configs/spdnet/spdnet_c32s3_l3ns3nr3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/configs/spdnet/spdnet_c32s3_l3ns3nr3.py -------------------------------------------------------------------------------- /dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/dist_test.sh -------------------------------------------------------------------------------- /dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/dist_train.sh -------------------------------------------------------------------------------- /docs/dataset_preparation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/docs/dataset_preparation.md -------------------------------------------------------------------------------- /docs/getting_started.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/docs/getting_started.md -------------------------------------------------------------------------------- /docs/installation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/docs/installation.md -------------------------------------------------------------------------------- /figs/brn.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/figs/brn.png -------------------------------------------------------------------------------- /figs/dcsfn.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/figs/dcsfn.png -------------------------------------------------------------------------------- /figs/ddn.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/figs/ddn.png -------------------------------------------------------------------------------- /figs/derainnet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/figs/derainnet.png -------------------------------------------------------------------------------- /figs/drdnet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/figs/drdnet.png -------------------------------------------------------------------------------- /figs/dual_gcn.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/figs/dual_gcn.png -------------------------------------------------------------------------------- /figs/ecnet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/figs/ecnet.png -------------------------------------------------------------------------------- /figs/id_cgan.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/figs/id_cgan.png -------------------------------------------------------------------------------- /figs/lpnet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/figs/lpnet.png -------------------------------------------------------------------------------- /figs/mardnet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/figs/mardnet.png -------------------------------------------------------------------------------- /figs/oucdnet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/figs/oucdnet.png -------------------------------------------------------------------------------- /figs/physical_model_guided.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/figs/physical_model_guided.png -------------------------------------------------------------------------------- /figs/prenet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/figs/prenet.png -------------------------------------------------------------------------------- /figs/rcdnet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/figs/rcdnet.png -------------------------------------------------------------------------------- /figs/rehen.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/figs/rehen.png -------------------------------------------------------------------------------- /figs/rescan.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/figs/rescan.png -------------------------------------------------------------------------------- /figs/rlnet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/figs/rlnet.png -------------------------------------------------------------------------------- /figs/spdnet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/figs/spdnet.png -------------------------------------------------------------------------------- /get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/get_flops.py -------------------------------------------------------------------------------- /get_model_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/get_model_info.py -------------------------------------------------------------------------------- /mmderain/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/__init__.py -------------------------------------------------------------------------------- /mmderain/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/apis/__init__.py -------------------------------------------------------------------------------- /mmderain/apis/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/apis/test.py -------------------------------------------------------------------------------- /mmderain/apis/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/apis/train.py -------------------------------------------------------------------------------- /mmderain/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/core/__init__.py -------------------------------------------------------------------------------- /mmderain/core/distributed_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/core/distributed_wrapper.py -------------------------------------------------------------------------------- /mmderain/core/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/core/evaluation/__init__.py -------------------------------------------------------------------------------- /mmderain/core/evaluation/eval_hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/core/evaluation/eval_hooks.py -------------------------------------------------------------------------------- /mmderain/core/evaluation/metric_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/core/evaluation/metric_utils.py -------------------------------------------------------------------------------- /mmderain/core/evaluation/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/core/evaluation/metrics.py -------------------------------------------------------------------------------- /mmderain/core/evaluation/niqe_pris_params.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/core/evaluation/niqe_pris_params.npz -------------------------------------------------------------------------------- /mmderain/core/hooks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/core/hooks/__init__.py -------------------------------------------------------------------------------- /mmderain/core/hooks/rlnet_hyperparam_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/core/hooks/rlnet_hyperparam_hook.py -------------------------------------------------------------------------------- /mmderain/core/initializers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/core/initializers/__init__.py -------------------------------------------------------------------------------- /mmderain/core/initializers/ecnet_initializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/core/initializers/ecnet_initializer.py -------------------------------------------------------------------------------- /mmderain/core/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/core/misc.py -------------------------------------------------------------------------------- /mmderain/core/optimizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/core/optimizer/__init__.py -------------------------------------------------------------------------------- /mmderain/core/optimizer/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/core/optimizer/builder.py -------------------------------------------------------------------------------- /mmderain/core/scheduler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/core/scheduler/__init__.py -------------------------------------------------------------------------------- /mmderain/core/scheduler/lr_updater.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/core/scheduler/lr_updater.py -------------------------------------------------------------------------------- /mmderain/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/datasets/__init__.py -------------------------------------------------------------------------------- /mmderain/datasets/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/datasets/base_dataset.py -------------------------------------------------------------------------------- /mmderain/datasets/base_derain_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/datasets/base_derain_dataset.py -------------------------------------------------------------------------------- /mmderain/datasets/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/datasets/builder.py -------------------------------------------------------------------------------- /mmderain/datasets/dataset_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/datasets/dataset_wrappers.py -------------------------------------------------------------------------------- /mmderain/datasets/derain_filename_matching_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/datasets/derain_filename_matching_dataset.py -------------------------------------------------------------------------------- /mmderain/datasets/derain_paired_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/datasets/derain_paired_dataset.py -------------------------------------------------------------------------------- /mmderain/datasets/derain_unpaired_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/datasets/derain_unpaired_dataset.py -------------------------------------------------------------------------------- /mmderain/datasets/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/datasets/pipelines/__init__.py -------------------------------------------------------------------------------- /mmderain/datasets/pipelines/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/datasets/pipelines/augmentation.py -------------------------------------------------------------------------------- /mmderain/datasets/pipelines/compose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/datasets/pipelines/compose.py -------------------------------------------------------------------------------- /mmderain/datasets/pipelines/crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/datasets/pipelines/crop.py -------------------------------------------------------------------------------- /mmderain/datasets/pipelines/formating.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/datasets/pipelines/formating.py -------------------------------------------------------------------------------- /mmderain/datasets/pipelines/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/datasets/pipelines/loading.py -------------------------------------------------------------------------------- /mmderain/datasets/pipelines/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/datasets/pipelines/normalization.py -------------------------------------------------------------------------------- /mmderain/datasets/pipelines/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/datasets/pipelines/utils.py -------------------------------------------------------------------------------- /mmderain/datasets/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/datasets/registry.py -------------------------------------------------------------------------------- /mmderain/datasets/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/datasets/samplers/__init__.py -------------------------------------------------------------------------------- /mmderain/datasets/samplers/distributed_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/datasets/samplers/distributed_sampler.py -------------------------------------------------------------------------------- /mmderain/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/__init__.py -------------------------------------------------------------------------------- /mmderain/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/backbones/__init__.py -------------------------------------------------------------------------------- /mmderain/models/backbones/brn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/backbones/brn.py -------------------------------------------------------------------------------- /mmderain/models/backbones/cgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/backbones/cgan.py -------------------------------------------------------------------------------- /mmderain/models/backbones/dcsfn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/backbones/dcsfn.py -------------------------------------------------------------------------------- /mmderain/models/backbones/ddn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/backbones/ddn.py -------------------------------------------------------------------------------- /mmderain/models/backbones/derainnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/backbones/derainnet.py -------------------------------------------------------------------------------- /mmderain/models/backbones/drdnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/backbones/drdnet.py -------------------------------------------------------------------------------- /mmderain/models/backbones/dual_gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/backbones/dual_gcn.py -------------------------------------------------------------------------------- /mmderain/models/backbones/ecnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/backbones/ecnet.py -------------------------------------------------------------------------------- /mmderain/models/backbones/lpnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/backbones/lpnet.py -------------------------------------------------------------------------------- /mmderain/models/backbones/mardnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/backbones/mardnet.py -------------------------------------------------------------------------------- /mmderain/models/backbones/oucdnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/backbones/oucdnet.py -------------------------------------------------------------------------------- /mmderain/models/backbones/physical_model_guided.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/backbones/physical_model_guided.py -------------------------------------------------------------------------------- /mmderain/models/backbones/prenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/backbones/prenet.py -------------------------------------------------------------------------------- /mmderain/models/backbones/rcdnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/backbones/rcdnet.py -------------------------------------------------------------------------------- /mmderain/models/backbones/rehen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/backbones/rehen.py -------------------------------------------------------------------------------- /mmderain/models/backbones/rescan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/backbones/rescan.py -------------------------------------------------------------------------------- /mmderain/models/backbones/rlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/backbones/rlnet.py -------------------------------------------------------------------------------- /mmderain/models/backbones/spdnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/backbones/spdnet.py -------------------------------------------------------------------------------- /mmderain/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/base.py -------------------------------------------------------------------------------- /mmderain/models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/builder.py -------------------------------------------------------------------------------- /mmderain/models/common/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/common/__init__.py -------------------------------------------------------------------------------- /mmderain/models/common/box_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/common/box_filter.py -------------------------------------------------------------------------------- /mmderain/models/common/guided_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/common/guided_filter.py -------------------------------------------------------------------------------- /mmderain/models/common/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/common/model_utils.py -------------------------------------------------------------------------------- /mmderain/models/common/priors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/common/priors.py -------------------------------------------------------------------------------- /mmderain/models/common/pyramids.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/common/pyramids.py -------------------------------------------------------------------------------- /mmderain/models/common/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/common/utils.py -------------------------------------------------------------------------------- /mmderain/models/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/layers/__init__.py -------------------------------------------------------------------------------- /mmderain/models/layers/ecnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/layers/ecnet.py -------------------------------------------------------------------------------- /mmderain/models/layers/mprnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/layers/mprnet.py -------------------------------------------------------------------------------- /mmderain/models/layers/rescan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/layers/rescan.py -------------------------------------------------------------------------------- /mmderain/models/layers/rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/layers/rnn.py -------------------------------------------------------------------------------- /mmderain/models/layers/se_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/layers/se_module.py -------------------------------------------------------------------------------- /mmderain/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/losses/__init__.py -------------------------------------------------------------------------------- /mmderain/models/losses/gan_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/losses/gan_loss.py -------------------------------------------------------------------------------- /mmderain/models/losses/gradient_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/losses/gradient_loss.py -------------------------------------------------------------------------------- /mmderain/models/losses/perceptual_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/losses/perceptual_loss.py -------------------------------------------------------------------------------- /mmderain/models/losses/pixelwise_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/losses/pixelwise_loss.py -------------------------------------------------------------------------------- /mmderain/models/losses/structural_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/losses/structural_loss.py -------------------------------------------------------------------------------- /mmderain/models/losses/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/losses/utils.py -------------------------------------------------------------------------------- /mmderain/models/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/registry.py -------------------------------------------------------------------------------- /mmderain/models/restorers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/restorers/__init__.py -------------------------------------------------------------------------------- /mmderain/models/restorers/basic_restorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/restorers/basic_restorer.py -------------------------------------------------------------------------------- /mmderain/models/restorers/cgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/restorers/cgan.py -------------------------------------------------------------------------------- /mmderain/models/restorers/ecnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/restorers/ecnet.py -------------------------------------------------------------------------------- /mmderain/models/restorers/lpnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/restorers/lpnet.py -------------------------------------------------------------------------------- /mmderain/models/restorers/multi_output_restorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/restorers/multi_output_restorer.py -------------------------------------------------------------------------------- /mmderain/models/restorers/multi_stage_restorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/restorers/multi_stage_restorer.py -------------------------------------------------------------------------------- /mmderain/models/restorers/physical_model_guided.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/restorers/physical_model_guided.py -------------------------------------------------------------------------------- /mmderain/models/restorers/rcdnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/restorers/rcdnet.py -------------------------------------------------------------------------------- /mmderain/models/restorers/rlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/restorers/rlnet.py -------------------------------------------------------------------------------- /mmderain/models/restorers/spdnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/models/restorers/spdnet.py -------------------------------------------------------------------------------- /mmderain/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/utils/__init__.py -------------------------------------------------------------------------------- /mmderain/utils/functools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/utils/functools.py -------------------------------------------------------------------------------- /mmderain/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/mmderain/utils/logger.py -------------------------------------------------------------------------------- /mmderain/version.py: -------------------------------------------------------------------------------- 1 | __version__ = '0.1.0' 2 | -------------------------------------------------------------------------------- /pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/pylintrc -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/requirements.txt -------------------------------------------------------------------------------- /requirements/runtime.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/requirements/runtime.txt -------------------------------------------------------------------------------- /requirements/tests.txt: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/test.py -------------------------------------------------------------------------------- /tools/publish_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/tools/publish_model.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biubiubiiu/derain-toolbox/HEAD/train.py --------------------------------------------------------------------------------