Downloads · 30 days
0
zaccharieramzi/UPDNet-knee-singlecoil-af4
UPDNet-knee-singlecoil-af4 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 18, 2021
Repo size
5 MB
Likes
0
Public
Click a slice to open those files.
.h55 MB · 100%
From the Hugging Face model README
tags:
This model can be used to reconstruct single coil fastMRI data with an acceleration factor of 4.
For more details, see https://arxiv.org/abs/2010.07290. This section is WIP.
This model can be used to reconstruct single coil knee data from Siemens scanner at acceleration factor 4. It cannot be used on multi-coil data.
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.updnet import UPDNet
model = UPDNet(
n_dual=1,
primal_only=True,
layers_n_channels=[16 * 2**i for i in range(3)],
)
kspace_size = [1, 320, 320]
inputs = [
tf.zeros(kspace_size + [1], dtype=tf.complex64), # kspace
tf.zeros(kspace_size, dtype=tf.complex64), # mask
]
model(inputs)
model.load_weights('model_weights.h5')
Using the model is then as simple as:
model([
kspace, # shape: [n_slices, n_rows, n_cols, 1]
mask, # shape: [n_slices, n_rows, n_cols]
])
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://arxiv.org/abs/2010.07290 for brain data. This section is WIP.
Not available
@inproceedings{Ramzi2020d,
archivePrefix = {arXiv},
arxivId = {2010.07290},
author = {Ramzi, Zaccharie and Ciuciu, Philippe and Starck, Jean-Luc},
booktitle = {ISMRM},
eprint = {2010.07290},
pages = {1--4},
title = {{XPDNet for MRI Reconstruction: an application to the 2020 fastMRI challenge}},
url = {http://arxiv.org/abs/2010.07290},
year = {2021}
}