A Neural networks toolbox with a focus on medical image analysis in tensorflow/keras for now.
Note: this is under ongoing development
SpatialTransformer
layer for N-D (dense and affine) spatial transforms, a vector integration layer VecInt
, sparse operations (e.g. SpatiallySparse_Dense
), and LocallyConnected3D
currently not included in keras
interpn
: N-D gridded interpolation, transform
: warp images, integrate_vec
: vector field integration, stack_models
: keras model stacking keras
training to help with understanding your fit, such as Dice measurements and volume-segmentation overlaps If you use this code, please cite:
Anatomical Priors in Convolutional Networks for Unsupervised Biomedical Segmentation
Adrian V. Dalca, John Guttag, Mert R. Sabuncu
CVPR 2018.
[ PDF | arxiv | bibtex ]
If you are using any of the sparse/imputation functions, please cite:
Unsupervised Data Imputation via Variational Inference of Deep Subspaces
Adrian V. Dalca, John Guttag, Mert R. Sabuncu
Arxiv preprint 2019
[ arxiv | bibtex ]
Please open an issue [preferred] or contact Adrian Dalca at adalca@csail.mit.edu for question related to neuron
.
Parts of neuron
were used in VoxelMorph and brainstorm, which we encourage you to check out!