zaccharieramzi commited on
Commit
ee72ff3
1 Parent(s): 6e104ff

added model card and model weights

Browse files
Files changed (2) hide show
  1. README.md +74 -0
  2. model_weights.h5 +3 -0
README.md ADDED
@@ -0,0 +1,74 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # PDNet-fastmri
2
+ ---
3
+ tags:
4
+ - TensorFlow
5
+ - MRI reconstruction
6
+ - MRI
7
+ datasets:
8
+ - fastMRI
9
+ ---
10
+
11
+ This model can be used to reconstruct single coil fastMRI 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 knee data from Siemens scanner 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 [fastMRI dataset](https://fastmri.org/dataset/).
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 [fastMRI dataset](https://fastmri.org/dataset/).
54
+
55
+ | Contrast | PD | PDFS |
56
+ |----------|-------|--------|
57
+ | PSNR | 34.2 | 30.06 |
58
+ | SSIM | 0.818 | 0.9554 |
59
+
60
+
61
+ ## Bibtex entry
62
+ ```
63
+ @article{ramzi2020benchmarking,
64
+ title={Benchmarking MRI reconstruction neural networks on large public datasets},
65
+ author={Ramzi, Zaccharie and Ciuciu, Philippe and Starck, Jean-Luc},
66
+ journal={Applied Sciences},
67
+ volume={10},
68
+ number={5},
69
+ pages={1816},
70
+ year={2020},
71
+ publisher={Multidisciplinary Digital Publishing Institute}
72
+ }
73
+ ```
74
+
model_weights.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cb24e1d6a0a2cf48dc15248384458668730e3ff99378499cd1d972ce06fc6855
3
+ size 1590472