Skip to content

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

License

Notifications You must be signed in to change notification settings

denti/AlexNet3D

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 

Repository files navigation

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

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages