/scratch-net

player detection and classification

Primary LanguageJupyter Notebook

scratch-net

Toy examples for object classification and classification

Dependencies:

  1. Python 3.6
  2. Pytorch 0.4.0
  3. matplotlib 2.2.2
  4. numpy 1.14.3
  5. Jupyter Notebook

Files:

  1. MNIST.ipynb is a simple CNN network for classification in mnist dataset.
  2. FCN.ipynb is mnist number classification using FCN network.
  3. mnist_module.mat is .mat file to save the training CNN model.