This is a single image deraining toolbox based on PyTorch and mmcv. Some benchmark algorithms for single image deraining are reimplemented in this repo, which are migrated from their official code or reproduced based on their papers (for those have no publicly available code). Experiments of these algorithms are carried out again, including evaluation on benchmark datasets (Rain200L, Rain200H, Rain800, Rain1200, and Rain1400) and tests of network complexity. The results are reported in the corresponding README documents.
- DerainNet (TIP' 2017) [paper] [code]
- DDN (CVPR' 2017) [paper] [code]
- RESCAN (ECCV' 2018) [paper] [code]
- PReNet (CVPR' 2019) [paper] [code]
- ReHEN (ACMMM' 2019) [paper] [code]
- LPNet (TNNLS' 2019) [paper] [code]
- Physical Model Guided ID (ICME' 2020) [paper] [code]
- MARDNet (ACCV' 2020) [paper] [code]
- OUCDNet (JSTSP' 2020) [paper] [code]
- ID-CGAN (TCSVT' 2020) [paper] [code]
- DCSFN (ACMMM' 2020) [paper] [code]
- BRN (TIP' 2020) [paper] [code]
- DRD-Net (CVPR' 2020) [paper] [code]
- RCDNet (CVPR' 2020) [paper] [code]
- DualGCN (AAAI' 2021) [paper] [code]
- SPDNet (ICCV' 2021) [paper] [code]
- DerainRLNet (CVPR' 2021) [paper] [code]
- ECNet (WACV' 2022) [paper] [code]
See getting started for the basic usage of derain-toolbox.
Portions of this project utilized other open-source works, the use of which is listed acknowledged in CREDITS.
If you've found an error in this project, please file an issue.
Patches are encouraged and may be submitted by forking this project and submitting a pull request. Since this project is still in its very early stages, if your change is substantial, please raise an issue first to discuss it.
Copyright 2021 Raymond Wong
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.