NiftyTorch: A Deep Learning framework for NeuroImaging

Abstract

We present NiftyTorch a Deep Learning Framework for NeuroImaging. The motivation behind the development of such a library is that there are scant amount of centralized tool for deploying 3D deep learning for NeuroImaging. In addition, most of the existing tools require expert technical knowledge in Deep Learning or programming, creating a barrier for entry. The goal is to provide a one stop package using which the users can perform classification tasks, Segmentation tasks and Image Transformation tasks. The intended audience are the members of NeuroImaging who would like to explore deep learning but have no background in coding. In this article we explore the capabilities of the framework, the performance of the framework and the future work for the framework.