pymedio 0.2.14

Install

The easiest way to install the package is through the following command:

pip install pymedio

Alternatively, this can be done via conda:

conda install -c conda-forge pymedio

To install from the source directory, clone the repo and run:

python setup.py install

To make the package as portable as possible, the package only comes with numpy installed by default.

If you want to load only DICOM images (using pydicom) install with:

pip install "pymedio[dicom]"

If you want to load DICOM and non-DICOM images (using SimpleITK or nibabel as a backend), install with:

pip install "pymedio[all]"

Basic Usage

Say you have a directory of DICOM images at the path dicom_dir, and you installed the package with dicom extras, then you can open it with:

import pymedio.dicom as miod
image = miod.DICOMImage.from_path("dicom_dir")

This uses pydicom as a backend to open the image à la dicom-numpy.

If you have a NIfTI image at image.nii, and you installed the package with all extras, you can open it with:

import pymedio.image as mioi
image = mioi.Image.from_path("image.nii")

In either case, you can proceed to work with the image data like a normal numpy array, e.g.,

image += 1.0
image *= image

Note that the image will have a affine attribute which stores the (affine) coordinate transformation matrix to/from scanner coordinates/voxels (even after applying numpy functions/operations to the image).

You can convert an either image class to a torch tensor—if you have it installed—like:

import torch
tensor = torch.as_tensor(image.torch_compatible())

If you want to save an image as a non-standard data type (e.g., a 16-bit floating point number), do so with:

# optionally convert the image to the desired type
f16_image = image.astype(np.float16)
f16_image.to_npz("image-f16.npz")

Then, in a more resource-constrained environment, you can install pymedio without any extras and run, e.g.,

from pymedio.base import ImageBase as Image
f16_image = Image.from_npz("image-f16.npz")
print(f16_image.affine)

To view the image in a standard reader following some processing (assuming all extras installed), you can then do, e.g.,

import pymedio.image as mioi
image = mioi.Image.from_npz("image-f16.npz")
image.astype(np.float32).save("image.nii.gz")

References