Pre-trained 3D LeNet-5
Updated 6 May 2021
Our implementation of 2D LeNet-5 model achieved 98.48% accuracy on the grey-scale MNIST test set after training on its train set. To transfer the learnable parameters from pre-trained 2D LeNet-5 (MNIST) to 3D one, we duplicated 2D filters (copying them repeatedly) through the third dimension. This is possible since a video or a 3D image can be converted into a sequence of image slices. In the training process, we expect that the 3D LeNet-5 learns patterns in each frame. This model has about 260,000 learnable parameters.
simply, call "lenet5TL3Dfun()" function.
Ebrahimi, Amir, et al. “Convolutional Neural Networks for Alzheimer’s Disease Detection on MRI Images.” Journal of Medical Imaging, vol. 8, no. 02, SPIE-Intl Soc Optical Eng, Apr. 2021, doi:10.1117/1.jmi.8.2.024503.
MATLAB Release Compatibility
Platform CompatibilityWindows macOS Linux
Inspired by: Deep Learning Network Analyzer for Neural Network Toolbox, Pre-trained 2D LeNet-5
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!Start Hunting!
Discover Live Editor
Create scripts with code, output, and formatted text in a single executable document.
The relevant paper is published.