zaccharieramzi commited on
Commit
71ea95a
1 Parent(s): ed276b4

added model card and model weights

Browse files
Files changed (2) hide show
  1. README.md +72 -0
  2. model_weights.h5 +3 -0
README.md ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # PDNet-OASIS
2
+ ---
3
+ tags:
4
+ - TensorFlow
5
+ - MRI reconstruction
6
+ - MRI
7
+ datasets:
8
+ - OASIS
9
+ ---
10
+
11
+ This model can be used to reconstruct single coil OASIS data with an acceleration factor of 4.
12
+
13
+ ## Model description
14
+ For more details, see https://www.mdpi.com/2076-3417/10/5/1816.
15
+ This section is WIP.
16
+
17
+ ## Intended uses and limitations
18
+ This model can be used to reconstruct single coil brain retrospective data from the OASIS database at acceleration factor 4.
19
+ It cannot be used on multi-coil data.
20
+
21
+ ## How to use
22
+ This model can be loaded using the following repo: https://github.com/zaccharieramzi/fastmri-reproducible-benchmark.
23
+ After cloning the repo, `git clone https://github.com/zaccharieramzi/fastmri-reproducible-benchmark`, you can install the package via `pip install fastmri-reproducible-benchmark`.
24
+ The framework is TensorFlow.
25
+
26
+ You can initialize and load the model weights as follows:
27
+ ```python
28
+ from fastmri_recon.models.functional_models.pdnet import pdnet
29
+
30
+ model = pdnet()
31
+ model.load_weights('model_weights.h5')
32
+ ```
33
+
34
+ Using the model is then as simple as:
35
+ ```python
36
+ model([
37
+ kspace, # shape: [n_slices, n_rows, n_cols, 1]
38
+ mask, # shape: [n_slices, n_rows, n_cols]
39
+ ])
40
+ ```
41
+
42
+ ## Limitations and bias
43
+ The limitations and bias of this model have not been properly investigated.
44
+
45
+ ## Training data
46
+ This model was trained using the [OASIS dataset](https://www.oasis-brains.org/).
47
+
48
+ ## Training procedure
49
+ The training procedure is described in https://www.mdpi.com/2076-3417/10/5/1816 for brain data.
50
+ This section is WIP.
51
+
52
+ ## Evaluation results
53
+ This model was evaluated using the [OASIS dataset](https://www.oasis-brains.org/).
54
+
55
+ - PSNR: 33.22
56
+ - SSIM: 0.910
57
+
58
+
59
+ ## Bibtex entry
60
+ ```
61
+ @article{ramzi2020benchmarking,
62
+ title={Benchmarking MRI reconstruction neural networks on large public datasets},
63
+ author={Ramzi, Zaccharie and Ciuciu, Philippe and Starck, Jean-Luc},
64
+ journal={Applied Sciences},
65
+ volume={10},
66
+ number={5},
67
+ pages={1816},
68
+ year={2020},
69
+ publisher={Multidisciplinary Digital Publishing Institute}
70
+ }
71
+ ```
72
+
model_weights.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:af483085067ea56288923c5a624a9e4e6ceb5fea6efa550c9fc59adc7b649133
3
+ size 1589624