Skip to content

README.md

Latest commit

 

History

History
17 lines (11 loc) · 760 Bytes

README.md

File metadata and controls

17 lines (11 loc) · 760 Bytes
Raw

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)