Downloads · 30 days
0
zaccharieramzi/CascadeNet-OASIS
CascadeNet-OASIS 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: - OASIS ---
Downloads · 30 days
0
Access
Public
Updated Dec 19, 2021
Repo size
1.8 MB
Likes
0
Public
Click a slice to open those files.
.h51.8 MB · 100%
From the Hugging Face model README
tags:
This model can be used to reconstruct single coil OASIS data with an acceleration factor of 4.
For more details, see https://www.mdpi.com/2076-3417/10/5/1816. This section is WIP.
This model can be used to reconstruct single coil brain retrospective data from the OASIS database 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:
from fastmri_recon.models.functional_models.cascading import cascade_net
model = cascade_net()
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 OASIS dataset.
The training procedure is described in https://www.mdpi.com/2076-3417/10/5/1816 for brain data. This section is WIP.
This model was evaluated using the OASIS dataset.
@article{ramzi2020benchmarking,
title={Benchmarking MRI reconstruction neural networks on large public datasets},
author={Ramzi, Zaccharie and Ciuciu, Philippe and Starck, Jean-Luc},
journal={Applied Sciences},
volume={10},
number={5},
pages={1816},
year={2020},
publisher={Multidisciplinary Digital Publishing Institute}
}