3d_convolution_neural_net_MNET

3d_convolution_neural_net_MNET copied to clipboard

3D Convolutional Neural Network

Download the dataset and helper function on kaggle following this link: https://www.kaggle.com/daavoo/3d-mnist

Structure

• keras_model.py: General model in keras for the keras architecture
• 3DkerasConv_example.ipynb: Jupyter notebook with example of keras usage (use after downloading the dataset)
• pytorch_model.py: General model in keras for the pytorch architecture
• 3DpytorchConv_example.ipynb: Jupyter notebook with example of pytorch usage (use after downloading the dataset)