zaccharieramzi commited on
Commit
073fc7a
1 Parent(s): 2047871

added model card and model weights

Browse files
Files changed (2) hide show
  1. README.md +75 -0
  2. model_weights.h5 +3 -0
README.md ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # KIKI-net-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.kiki_sep import full_kiki_net
29
+ from fastmri_recon.models.utils.non_linearities import lrelu
30
+
31
+
32
+ model = full_kiki_net(n_convs=16, n_filters=48, activation=lrelu)
33
+ model.load_weights('model_weights.h5')
34
+
35
+ ```
36
+
37
+ Using the model is then as simple as:
38
+ ```python
39
+ model([
40
+ kspace, # shape: [n_slices, n_rows, n_cols, 1]
41
+ mask, # shape: [n_slices, n_rows, n_cols]
42
+ ])
43
+ ```
44
+
45
+ ## Limitations and bias
46
+ The limitations and bias of this model have not been properly investigated.
47
+
48
+ ## Training data
49
+ This model was trained using the [OASIS dataset](https://www.oasis-brains.org/).
50
+
51
+ ## Training procedure
52
+ The training procedure is described in https://www.mdpi.com/2076-3417/10/5/1816 for brain data.
53
+ This section is WIP.
54
+
55
+ ## Evaluation results
56
+ This model was evaluated using the [OASIS dataset](https://www.oasis-brains.org/).
57
+
58
+ - PSNR: 30.08
59
+ - SSIM: 0.853
60
+
61
+
62
+ ## Bibtex entry
63
+ ```
64
+ @article{ramzi2020benchmarking,
65
+ title={Benchmarking MRI reconstruction neural networks on large public datasets},
66
+ author={Ramzi, Zaccharie and Ciuciu, Philippe and Starck, Jean-Luc},
67
+ journal={Applied Sciences},
68
+ volume={10},
69
+ number={5},
70
+ pages={1816},
71
+ year={2020},
72
+ publisher={Multidisciplinary Digital Publishing Institute}
73
+ }
74
+ ```
75
+
model_weights.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6818c4bd7f3ad6600c74db33fc35061c5614735015859a2b21fc57296fbbd061
3
+ size 7770568