zaccharieramzi commited on
Commit
c19f07f
1 Parent(s): 99cf0bf

added model card and model weights

Browse files
Files changed (2) hide show
  1. README.md +83 -0
  2. model_weights.h5 +3 -0
README.md ADDED
@@ -0,0 +1,83 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # UPDNet-knee-singlecoil-af4
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://arxiv.org/abs/2010.07290.
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
+ import tensorflow as tf
29
+
30
+ from fastmri_recon.models.subclassed_models.updnet import UPDNet
31
+
32
+
33
+ model = UPDNet(
34
+ n_dual=1,
35
+ primal_only=True,
36
+ layers_n_channels=[16 * 2**i for i in range(3)],
37
+ )
38
+ kspace_size = [1, 320, 320]
39
+ inputs = [
40
+ tf.zeros(kspace_size + [1], dtype=tf.complex64), # kspace
41
+ tf.zeros(kspace_size, dtype=tf.complex64), # mask
42
+ ]
43
+ model(inputs)
44
+ model.load_weights('model_weights.h5')
45
+ ```
46
+
47
+ Using the model is then as simple as:
48
+ ```python
49
+ model([
50
+ kspace, # shape: [n_slices, n_rows, n_cols, 1]
51
+ mask, # shape: [n_slices, n_rows, n_cols]
52
+ ])
53
+ ```
54
+
55
+ ## Limitations and bias
56
+ The limitations and bias of this model have not been properly investigated.
57
+
58
+ ## Training data
59
+ This model was trained using the [fastMRI dataset](https://fastmri.org/dataset/).
60
+
61
+ ## Training procedure
62
+ The training procedure is described in https://arxiv.org/abs/2010.07290 for brain data.
63
+ This section is WIP.
64
+
65
+ ## Evaluation results
66
+ Not available
67
+
68
+
69
+ ## Bibtex entry
70
+ ```
71
+ @inproceedings{Ramzi2020d,
72
+ archivePrefix = {arXiv},
73
+ arxivId = {2010.07290},
74
+ author = {Ramzi, Zaccharie and Ciuciu, Philippe and Starck, Jean-Luc},
75
+ booktitle = {ISMRM},
76
+ eprint = {2010.07290},
77
+ pages = {1--4},
78
+ title = {{XPDNet for MRI Reconstruction: an application to the 2020 fastMRI challenge}},
79
+ url = {http://arxiv.org/abs/2010.07290},
80
+ year = {2021}
81
+ }
82
+ ```
83
+
model_weights.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a11c976f9e27e256e20c59c6b9a26f3b561ac1dd6bef5801e588280c69b35cd5
3
+ size 4999888