This repository contains utilities for handling data in medical image analysis that are not specific to certain tasks.
So far it consists of 2 major parts:
An abstract Dataset API built on top of torch.utils.data.Dataset and torchio.data.Subject as well as general transforms for this kind of data.
To use the dataset classes, you basically only need to implement the parse_subjects method to return a list of samples and everything else will work automatically.
You will automatically get image statistics such as median spacing or median shape. For label statistics, you either need to subclass the AbstractDiscreteLabelDataset or implement the get_single_label_stats and aggregate_label_stats methods.
All transforms work on torchio.data.Subjects and can be passed to the datasets as optional parameters. You can also pass "default" as a parameter to use the default transforms.
Pull requests for other common utilities are highly welcomed.
This project can be installed either from PyPI or by cloning the repository from GitHub.
For an install of published packages, use the command
pip install medical-data
To install from the (cloned) repository, use the command
pip install PATH/TO/medical-data
You can also add -e to the command to make an editable install in case you want to modify the code.
You can also install the package directly from GitHub by running
pip install git+https://github.com/justusschock/medical-data.git