Skip to main content

Data pipelines for the fastMRI dataset in TensorFlow.

Project description


Build Status

Built around the API, tf-fastmri-data offers reliable, unit-tested, datasets for the fastMRI dataset.


Currently, you need to install the package from source:

git clone
cd tf-fastmri-data
pip install .

Example use

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

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):

TensorBoard reco


If you use the fastMRI data or this code in your research, please consider citing the fastMRI dataset paper:

  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},

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

tf-fastmri-data-0.0.6.tar.gz (18.0 kB view hashes)

Uploaded source

Built Distribution

tf_fastmri_data-0.0.6-py3-none-any.whl (22.6 kB view hashes)

Uploaded py3

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor NVIDIA NVIDIA PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page