├── DRMI_evaluation.py ├── DRMI_train.py ├── LICENSE ├── README.md ├── configure.py ├── data ├── USAF_Images │ ├── USAF_b_DRMI.png │ ├── USAF_b_NAFNet.png │ ├── USAF_b_meta.png │ ├── USAF_w_DRMI.png │ ├── USAF_w_NAFNet.png │ └── USAF_w_meta.png ├── data_util.py └── datasets.py ├── evaluation.py ├── figures ├── Fig1.png └── qualitative_results.png ├── loss ├── loss_util.py └── losses.py ├── models ├── DRMI.py ├── HINet.py ├── MIRNetv2.py ├── MPRNet.py ├── Restormer.py ├── SFNet.py ├── __init__.py ├── discriminator.py └── model_utils.py ├── requirements.txt ├── schedulers.py ├── train.py └── utils.py /DRMI_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/DRMI_evaluation.py -------------------------------------------------------------------------------- /DRMI_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/DRMI_train.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/README.md -------------------------------------------------------------------------------- /configure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/configure.py -------------------------------------------------------------------------------- /data/USAF_Images/USAF_b_DRMI.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/data/USAF_Images/USAF_b_DRMI.png -------------------------------------------------------------------------------- /data/USAF_Images/USAF_b_NAFNet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/data/USAF_Images/USAF_b_NAFNet.png -------------------------------------------------------------------------------- /data/USAF_Images/USAF_b_meta.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/data/USAF_Images/USAF_b_meta.png -------------------------------------------------------------------------------- /data/USAF_Images/USAF_w_DRMI.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/data/USAF_Images/USAF_w_DRMI.png -------------------------------------------------------------------------------- /data/USAF_Images/USAF_w_NAFNet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/data/USAF_Images/USAF_w_NAFNet.png -------------------------------------------------------------------------------- /data/USAF_Images/USAF_w_meta.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/data/USAF_Images/USAF_w_meta.png -------------------------------------------------------------------------------- /data/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/data/data_util.py -------------------------------------------------------------------------------- /data/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/data/datasets.py -------------------------------------------------------------------------------- /evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/evaluation.py -------------------------------------------------------------------------------- /figures/Fig1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/figures/Fig1.png -------------------------------------------------------------------------------- /figures/qualitative_results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/figures/qualitative_results.png -------------------------------------------------------------------------------- /loss/loss_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/loss/loss_util.py -------------------------------------------------------------------------------- /loss/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/loss/losses.py -------------------------------------------------------------------------------- /models/DRMI.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/models/DRMI.py -------------------------------------------------------------------------------- /models/HINet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/models/HINet.py -------------------------------------------------------------------------------- /models/MIRNetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/models/MIRNetv2.py -------------------------------------------------------------------------------- /models/MPRNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/models/MPRNet.py -------------------------------------------------------------------------------- /models/Restormer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/models/Restormer.py -------------------------------------------------------------------------------- /models/SFNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/models/SFNet.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/models/discriminator.py -------------------------------------------------------------------------------- /models/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/models/model_utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/requirements.txt -------------------------------------------------------------------------------- /schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/schedulers.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhy258/EIDL_DRMI/HEAD/utils.py --------------------------------------------------------------------------------