Skip to content

Latest commit

 

History

History
14 lines (10 loc) · 1.04 KB

README.md

File metadata and controls

14 lines (10 loc) · 1.04 KB
Raw

AlexNet_3dConv

TensorFlow implementation of AlexNet(2012) by Alex Krizhevsky, with 3D convolutiional layers.

3D AlexNet

3D AlexNet Network with a standart AlexNet architecture, but it has 3D instead 2D filters on each Conv and Pool layers.

Standart AlexNet

Standart AlexNet

To fit this model into memory, model training proccess can be refactored with @OpenAI's Gradient Checkpointing algorithm https://github.com/openai/gradient-checkpointing

  • 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 (or https://github.com/openai/gradient-checkpointing for example).

  • 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.