Built around the tf.data
API, tf-fastmri-data
offers reliable, unit-tested, datasets for the fastMRI dataset.
Currently, you need to install the package from source:
git clone https://github.com/zaccharieramzi/tf-fastmri-data.git
cd tf-fastmri-data
pip install .
from tf_fastmri_data.datasets.cartesian import CartesianFastMRIDatasetBuilder
train_dataset = CartesianFastMRIDatasetBuilder(path='/path/to/singlecoil_train').preprocessed_ds
To download the data, you need to consent to the fastMRI terms listed here. Afterwards, you should receive an email with data download links.
You can then use the environment variable FASTMRI_DATA_DIR
to indicate where your fastMRI is.
This will allow you to not have to specify the path when instantiating a FastMRIDatasetBuilder
.
The PyTorch equivalent of this library is simply the official fastMRI repository. In particular, the data folder is where you find the data utils.
You can run the benchmark script with the following command:
FASTMRI_DATA_DIR=/path/to/fastmri python benchmark.py
Currently the benchmark gives the following output:
Multi coil with tfio loading (random slice): 0.369743709564209s per-file.
Single coil with tfio loading (random slice): 0.02855397939682007s per-file.
Multi coil with h5py loading (random slice, without preprocessing): 0.010439331208042165s per-file.
Single coil with h5py loading (random slice, without preprocessing): 0.0015996736497735258s per-file.
Single coil training with tfio loading: 0.04578723907470703s per-step.
You can also see the recommendation of TensorBoard regarding the single coil dataset (with a very simple model):
If you use the fastMRI data or this code in your research, please consider citing the fastMRI dataset paper:
@inproceedings{zbontar2018fastMRI,
title={{fastMRI}: An Open Dataset and Benchmarks for Accelerated {MRI}},
author={Jure Zbontar and Florian Knoll and Anuroop Sriram and Matthew J. Muckley and Mary Bruno and Aaron Defazio and Marc Parente and Krzysztof J. Geras and Joe Katsnelson and Hersh Chandarana and Zizhao Zhang and Michal Drozdzal and Adriana Romero and Michael Rabbat and Pascal Vincent and James Pinkerton and Duo Wang and Nafissa Yakubova and Erich Owens and C. Lawrence Zitnick and Michael P. Recht and Daniel K. Sodickson and Yvonne W. Lui},
journal = {ArXiv e-prints},
archivePrefix = "arXiv",
eprint = {1811.08839},
year={2018}
}