NiftyNet icon indicating copy to clipboard operation
NiftyNet copied to clipboard

Switch BatchNorm implementation to rely on the official tf.layers.batch_normalization

Open tvercaut opened this issue 7 years ago • 1 comments

Batch Normalisation is officially supported in the main TensorFlow repo: https://www.tensorflow.org/api_docs/python/tf/layers/batch_normalization

It would be beneficial to rely on this rather than our own implementation: https://github.com/NifTK/NiftyNet/blob/dev/niftynet/layer/bn.py

Note that other users have done that already, see e.g.: https://github.com/gift-surg/fetal_brain_seg/blob/master/NiftyNet/niftynet/layer/bn.py vs. https://github.com/gift-surg/fetal_brain_seg/blob/master/NiftyNet/niftynet/layer/bn_backup.py

tvercaut avatar Nov 12 '18 12:11 tvercaut

This relates also to #280

tvercaut avatar Nov 12 '18 12:11 tvercaut