Skip to main content

Tools for loading, augmenting and writing 3D medical images on PyTorch.

Project description

TorchIO

PyPI downloads PyPI version Google Colab Build status Coverage status Code quality Code maintainability Slack


🎉 News: the paper is out! 🎉

See the Credits section below for more information.


Original Random blur
Original Random blur
Random flip Random noise
Random flip Random noise
Random affine transformation Random elastic transformation
Random affine transformation Random elastic transformation
Random bias field artifact Random motion artifact
Random bias field artifact Random motion artifact
Random spike artifact Random ghosting artifact
Random spike artifact Random ghosting artifact

TorchIO is a Python package containing a set of tools to efficiently read, sample and write 3D medical images in deep learning applications written in PyTorch, including intensity and spatial transforms for data augmentation and preprocessing. Transforms include typical computer vision operations such as random affine transformations and also domain-specific ones such as simulation of intensity artifacts due to MRI magnetic field inhomogeneity or k-space motion artifacts.

This package has been greatly inspired by NiftyNet.

Credits

If you like this repository, please click on Star!

If you use this package for your research, please cite the paper:

Pérez-García et al., 2020, TorchIO: a Python library for efficient loading, preprocessing, augmentation and patch-based sampling of medical images in deep learning.

BibTeX entry:

@misc{fern2020torchio,
    title={TorchIO: a Python library for efficient loading, preprocessing, augmentation and patch-based sampling of medical images in deep learning},
    author={Fernando Pérez-García and Rachel Sparks and Sebastien Ourselin},
    year={2020},
    eprint={2003.04696},
    archivePrefix={arXiv},
    primaryClass={eess.IV}
}

Documentation

The documentation is hosted on Read the Docs. It is a work in progress, but some classes such as ImagesDataset are already fairly well documented.

======= History

0.13.0 (24-02-2020)

  • Add Subject class
  • Add random blur transform
  • Add lambda transform
  • Add random patches swapping transform
  • Add MRI k-space ghosting artefact augmentation

0.12.0 (21-01-2020)

  • Add ToCanonical transform
  • Add CenterCropOrPad transform

0.11.0 (15-01-2020)

  • Add Resample transform

0.10.0 (15-01-2020)

  • Add Pad transform
  • Add Crop transform

0.9.0 (14-01-2020)

  • Add CLI tool to transform an image from file

0.8.0 (11-01-2020)

  • Add Image class

0.7.0 (02-01-2020)

  • Make transforms use PyTorch tensors consistently

0.6.0 (02-01-2020)

  • Add support for NRRD

0.5.0 (01-01-2020)

  • Add bias field transform

0.4.0 (29-12-2019)

  • Add MRI k-space motion artefact augmentation

0.3.0 (21-12-2019)

  • Add Rescale transform
  • Add support for multimodal data and missing modalities

0.2.0 (2019-12-06)

  • First release on PyPI.

Project details


Release history Release notifications | RSS feed

Download files

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

Source Distribution

torchio-0.13.16.tar.gz (21.8 MB view details)

Uploaded Source

Built Distribution

torchio-0.13.16-py2.py3-none-any.whl (56.7 kB view details)

Uploaded Python 2 Python 3

File details

Details for the file torchio-0.13.16.tar.gz.

File metadata

  • Download URL: torchio-0.13.16.tar.gz
  • Upload date:
  • Size: 21.8 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.1.1 pkginfo/1.5.0.1 requests/2.23.0 setuptools/46.0.0 requests-toolbelt/0.9.1 tqdm/4.43.0 CPython/3.7.1

File hashes

Hashes for torchio-0.13.16.tar.gz
Algorithm Hash digest
SHA256 e3920094ca1cf2609dfcea8cbfbcd5432cdbe7ccd2e1232442ec176737473b43
MD5 e208794341b73e62447f0330d14d2174
BLAKE2b-256 32e009f165443d80453dc4f9bc24feb8a822a97aa2349f323bca8e46e88e7968

See more details on using hashes here.

File details

Details for the file torchio-0.13.16-py2.py3-none-any.whl.

File metadata

  • Download URL: torchio-0.13.16-py2.py3-none-any.whl
  • Upload date:
  • Size: 56.7 kB
  • Tags: Python 2, Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.1.1 pkginfo/1.5.0.1 requests/2.23.0 setuptools/46.0.0 requests-toolbelt/0.9.1 tqdm/4.43.0 CPython/3.7.1

File hashes

Hashes for torchio-0.13.16-py2.py3-none-any.whl
Algorithm Hash digest
SHA256 9e4c963ed065f00f955e8728879747d2ed6037b2eb660d1d8f1317c5f3d1129e
MD5 1a9e1cc98e7e4d75a8268595e1a8ab47
BLAKE2b-256 21072ae3b0d90f918de512cc925bfdf229f486b68f8f731741b9bdafb77ce083

See more details on using hashes here.

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