miki998 / 3d_convolution_neural_net_MNET

Keras Implementation of 3d Convolutional Neural Network

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

3D Convolutional Neural Network

ko-fi

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)

About

Keras Implementation of 3d Convolutional Neural Network


Languages

Language:Jupyter Notebook 92.7%Language:Python 7.3%