NiftyNet is a TensorFlow-based open-source convolutional neural networks (CNN) platform for research in medical image analysis and image-guided therapy. NiftyNet's modular structure is designed for sharing networks and pre-trained models. Using this modular structure you can:
- Get started with established pre-trained networks using built-in tools
- Adapt existing networks to your imaging data
- Quickly build new solutions to your own image analysis problems
NiftyNet is a consortium of research groups (WEISS -- Wellcome EPSRC Centre for Interventional and Surgical Sciences, CMIC -- Centre for Medical Image Computing, HIG -- High-dimensional Imaging Group), where WEISS acts as the consortium lead.
NiftyNet currently supports medical image segmentation and generative adversarial networks. NiftyNet is not intended for clinical use. Other features of NiftyNet include:
- Easy-to-customise interfaces of network components
- Sharing networks and pretrained models
- Support for 2-D, 2.5-D, 3-D, 4-D inputs*
- Efficient discriminative training with multiple-GPU support
- Implementation of recent networks (HighRes3DNet, 3D U-net, V-net, DeepMedic)
- Comprehensive evaluation metrics for medical image segmentation
*2.5-D: volumetric images processed as a stack of 2D slices; 4-D: co-registered multi-modal 3D volumes
pip install niftynet
(This will install the GPU version of NiftyNet.)
Please follow the links for demos and network (re-)implementations.
Feature requests and bug reports are collected on Issues.
Contributors are encouraged to take a look at CONTRIBUTING.md.
If you use NiftyNet, please cite the following paper:
@InProceedings{niftynet17,
author = {Li, Wenqi and Wang, Guotai and Fidon, Lucas and Ourselin, Sebastien and Cardoso, M. Jorge and Vercauteren, Tom},
title = {On the Compactness, Efficiency, and Representation of 3D Convolutional Networks: Brain Parcellation as a Pretext Task},
booktitle = {International Conference on Information Processing in Medical Imaging (IPMI)},
year = {2017}
}
Copyright 2017 University College London and the NiftyNet Contributors. NiftyNet is released under the Apache License, Version 2.0. Please see the LICENSE file for details.
This project is grateful for the support from the Wellcome Trust, the Engineering and Physical Sciences Research Council (EPSRC), the National Institute for Health Research (NIHR), the Department of Health (DoH), University College London (UCL), the Science and Engineering South Consortium (SES), the STFC Rutherford-Appleton Laboratory, and NVIDIA.