Skip to main content

A collection of TensorFlow add-ons for computational MRI.

Project description

TFMRI logo

pypi build docs doi

TensorFlow MRI is a library of TensorFlow operators for computational MRI which includes:

  • A fast, native non-uniform fast Fourier transform (NUFFT) operator (see also TensorFlow NUFFT).

  • MR image reconstruction functions, which support parallel imaging, compressed sensing, machine learning and partial Fourier methods.

  • Common linear and nonlinear operators, such as Fourier operators and regularizers, to aid in the development of image reconstruction techniques.

  • Multicoil imaging operators, such as coil combination, coil compression and estimation of coil sensitivity maps.

  • Calculation of non-Cartesian k-space trajectories and sampling density estimation.

  • A collection of Keras objects including models, layers, metrics, loss functions and callbacks for rapid development of neural networks.

  • Many other differentiable operators for common tasks such as array manipulation and image/signal processing.

The library has a Python interface and is mostly written in Python. However, computations are efficiently performed by the TensorFlow backend (implemented in C++/CUDA), which brings together the ease of use and fast prototyping of Python with the speed and efficiency of optimized lower-level implementations.

Being an extension of TensorFlow, TensorFlow MRI integrates seamlessly in ML applications. No additional interfacing is needed to include a SENSE operator within a neural network, or to use a trained prior as part of an iterative reconstruction. Therefore, the gap between ML and non-ML components of image processing pipelines is eliminated.

Whether an application involves ML or not, TensorFlow MRI operators can take full advantage of the TensorFlow framework, with capabilities including automatic differentiation, multi-device support (CPUs and GPUs), automatic device placement and copying of tensor data, and conversion to fast, serializable graphs.

Installation

You can install TensorFlow MRI with pip:

$ pip install tensorflow-mri

Note that only Linux is currently supported.

TensorFlow Compatibility

Each TensorFlow MRI release is compiled against a specific version of TensorFlow. To ensure compatibility, it is recommended to install matching versions of TensorFlow and TensorFlow MRI according to the table below.

TensorFlow MRI Version

TensorFlow Compatibility

Release Date

v0.16.0

v2.8.x

Apr 13, 2022

v0.15.0

v2.8.x

Apr 1, 2022

v0.14.0

v2.8.x

Mar 29, 2022

v0.13.0

v2.8.x

Mar 15, 2022

v0.12.0

v2.8.x

Mar 14, 2022

v0.11.0

v2.8.x

Mar 10, 2022

v0.10.0

v2.8.x

Mar 3, 2022

v0.9.0

v2.7.x

Dec 3, 2021

v0.8.0

v2.7.x

Nov 11, 2021

v0.7.0

v2.6.x

Nov 3, 2021

v0.6.2

v2.6.x

Oct 13, 2021

v0.6.1

v2.6.x

Sep 30, 2021

v0.6.0

v2.6.x

Sep 28, 2021

v0.5.0

v2.6.x

Aug 29, 2021

v0.4.0

v2.6.x

Aug 18, 2021

Documentation

Visit the docs for the API reference and examples of usage.

Contributions

If you use this package and something does not work as you expected, please file an issue describing your problem. We will do our best to help.

Contributions are very welcome. Please create a pull request if you would like to make a contribution.

Citation

If you find this software useful in your work, please cite us.

FAQ

When trying to install TensorFlow MRI, I get an error about OpenEXR which includes: ``OpenEXR.cpp:36:10: fatal error: ImathBox.h: No such file or directory``. What do I do?

OpenEXR is needed by TensorFlow Graphics, which is a dependency of TensorFlow MRI. This issue can be fixed by installing the OpenEXR library. On Debian/Ubuntu:

$ apt install libopenexr-dev

Project details


Download files

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

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distributions

tensorflow_mri-0.16.0-cp310-cp310-manylinux_2_12_x86_64.manylinux2010_x86_64.whl (198.4 kB view hashes)

Uploaded CPython 3.10 manylinux: glibc 2.12+ x86-64

tensorflow_mri-0.16.0-cp39-cp39-manylinux_2_12_x86_64.manylinux2010_x86_64.whl (198.4 kB view hashes)

Uploaded CPython 3.9 manylinux: glibc 2.12+ x86-64

tensorflow_mri-0.16.0-cp38-cp38-manylinux_2_12_x86_64.manylinux2010_x86_64.whl (198.4 kB view hashes)

Uploaded CPython 3.8 manylinux: glibc 2.12+ x86-64

tensorflow_mri-0.16.0-cp37-cp37m-manylinux_2_12_x86_64.manylinux2010_x86_64.whl (198.4 kB view hashes)

Uploaded CPython 3.7m manylinux: glibc 2.12+ x86-64

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page