Add MEDL model

Author: georgeyiasemis

direct/nn/medl/__init__.py

@@ -0,0 +1,9 @@
+# Copyright (c) DIRECT Contributors
+
+"""Contains the implementation of the MEDL network [1]_ for MRI reconstruction.
+
+References
+----------
+.. [1] Qiao, X., Huang, Y., Li, W.: MEDL‐Net: A model‐based neural network for MRI reconstruction with enhanced deep 
+    learned regularizers. Magnetic Resonance in Med. 89, 2062–2075 (2023). https://doi.org/10.1002/mrm.29575
+"""

direct/nn/medl/config.py

@@ -0,0 +1,28 @@
+# Copyright (c) DIRECT Contributors
+
+"""Contains the configuration of MEDL models."""
+
+
+from dataclasses import dataclass
+
+from direct.config.defaults import ModelConfig
+
+
+@dataclass
+class MEDLConfig(ModelConfig):
+    iterations: int = (4,)
+    num_layers: int = 3
+    unet_num_filters: int = 18
+    unet_num_pool_layers: int = 4
+    unet_dropout: float = 0.0
+    unet_norm: bool = False
+
+
+@dataclass
+class MEDL2DConfig(MEDLConfig):
+    pass
+
+
+@dataclass
+class MEDL3DConfig(MEDLConfig):
+    pass