Downloads · 30 days
0
zaccharieramzi/NCPDNet-multicoil-spiral
NCPDNet-multicoil-spiral is a machine learning model from zaccharieramzi. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
--- tags: - TensorFlow - MRI reconstruction - MRI datasets: - fastMRI ---
Downloads · 30 days
0
Access
Public
Updated Dec 19, 2021
Repo size
770 KB
Likes
0
Public
Click a slice to open those files.
.h5770 KB · 100%
From the Hugging Face model README
tags:
This is a non-Cartesian multicoil MRI reconstruction model for spiral trajectories at acceleration factor 4. The model uses 10 iterations and a small vanilla CNN.
For more details, see https://hal.inria.fr/hal-03188997. This section is WIP.
This model can be used to reconstruct multicoil knee data from Siemens scanner at acceleration factor 4 in a spiral acquisition setting.
This model can be loaded using the following repo: https://github.com/zaccharieramzi/fastmri-reproducible-benchmark.
After cloning the repo, git clone https://github.com/zaccharieramzi/fastmri-reproducible-benchmark, you can install the package via pip install fastmri-reproducible-benchmark.
The framework is TensorFlow.
You can initialize and load the model weights as follows:
import tensorflow as tf
from fastmri_recon.models.subclassed_models.ncpdnet import NCPDNet
model = NCPDNet(
multicoil=True,
im_size=(640, 400),
dcomp=True,
refine_smaps=True,
)
kspace_shape = 1
inputs = [
tf.zeros([1, 1, kspace_shape, 1], dtype=tf.complex64),
tf.zeros([1, 2, kspace_shape], dtype=tf.float32),
tf.zeros([1, 1, 640, 320], dtype=tf.complex64),
(tf.constant([320]), tf.ones([1, kspace_shape], dtype=tf.float32)),
]
model(inputs)
model.load_weights('model_weights.h5')
Using the model is then as simple as:
model([
kspace, # shape: [n_slices, n_coils, n_kspace_samples, 1]
traj, # shape: [n_slices, n_coils, 2, n_kspace_samples]
smaps, # shape: [n_slices, n_coils, n_kspace_samples, n_coils]
(
output_shape, # shape: [n_slices, 1]
dcomp, # shape: [n_slices, n_kspace_samples]
)
])
The limitations and bias of this model have not been properly investigated.
This model was trained using the fastMRI dataset.
The training procedure is described in https://hal.inria.fr/hal-03188997. This section is WIP.
On the fastMRI validation dataset:
@unpublished{ramzi:hal-03188997,
TITLE = {{NC-PDNet: a Density-Compensated Unrolled Network for 2D and 3D non-Cartesian MRI Reconstruction}},
AUTHOR = {Ramzi, Zaccharie and G R, Chaithya and Starck, Jean-Luc and Ciuciu, Philippe},
YEAR = {2021},
MONTH = Sep,
}