Skip to main content

Data pipelines for the fastMRI dataset in TensorFlow.

Project description

tf-fastmri-data

Build Status

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

Installation

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 .

Example use

from tf_fastmri_data.datasets.cartesian import CartesianFastMRIDatasetBuilder

train_dataset = CartesianFastMRIDatasetBuilder(path='/path/to/singlecoil_train').preprocessed_ds

Data

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.

PyTorch

The PyTorch equivalent of this library is simply the official fastMRI repository. In particular, the data folder is where you find the data utils.

Benchmark

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

TensorBoard reco

Citation

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

Project details


Download files

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

Files for tf-fastmri-data, version 0.0.6
Filename, size File type Python version Upload date Hashes
Filename, size tf_fastmri_data-0.0.6-py3-none-any.whl (22.6 kB) File type Wheel Python version py3 Upload date Hashes View
Filename, size tf-fastmri-data-0.0.6.tar.gz (18.0 kB) File type Source Python version None Upload date Hashes View

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