This package simply provides appropriate dataset and transforms classes for NIfTI images
for use with PyTorch or PyTorch wrappers
** Note that this is an alpha release. If you have feedback or problems, please submit an issue (it is very appreciated) **
- nibabel >= 2.3.1
- numpy >= 1.15.4
- pillow >= 5.3.0
- torch >= 1.0.0
- torchvision >= 0.2.1
pip install git+git://github.com/jcreinhold/niftidataset.git
In addition to the above small tutorial, there is consolidated documentation here.
Unit tests can be run from the main directory as follows:
nosetests -v tests