I have written
torchio, a Python package with tools for patch-based training and inference of 3D medical images and multiple transforms for data augmentation typically used in the field. The
torchio.transforms API is similar to
torchvision.transforms. All the processing is done using PyTorch, NumPy and ITK.
The library is still very immature, so contributions and feedback are very welcome If you have the opportunity to try it out, please let me know what you think and report any issues on GitHub or here.