TensorFlow implementation of AlexNet(2012) by Alex Krizhevsky, with 3D convolutiional layers.
Network with a standart AlexNet architecture, but it has 3D instead 2D filters on each
Conv
and Pool
layers.
-
Note: This model needs a lot of GPU memory. Training session will not starts on 1 GPU without additional data separation or code's parallelization.
-
Note2: This model doesn't pretend to be the
SilverBullet
in 3D image recognition (like AlexNet was in 2D). It's just an example of 3D convolutional model in TensorFlow.