A PyTorch wrapper for Neuroimaging.
nitorch makes building deep learning classifiers for your 3D neuroimaging data (e.g. MRI) easy. It provides a high-level training framework, data augmentation schemes on multi-dimensional input and many other helpful tools.
Use our tutorial to learn how to use nitorch for building a deep learning Alzheimer's disease classifier based on the ADNI dataset: