denti / AlexNet3D

This is implementation of AlexNet(2012) with 3D Convolution on TensorFlow (AlexNet 3D).

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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.

About

This is implementation of AlexNet(2012) with 3D Convolution on TensorFlow (AlexNet 3D).

License:MIT License


Languages

Language:Python 100.0%