/LeNet5

Pure numpy implementation of LeNet5, to help you understand how CNN works.

Primary LanguagePythonMIT LicenseMIT

LeNet5

Pure numpy implementation of LeNet5, to help you understand how CNN works.

prerequisite

  1. python
  2. numpy
  3. opencv
pip install numpy
pip install opencv-python

usage

  1. Get mnist datasets and put it into the project root directory, then decompress it.
  2. train
python train.py
  1. test
python test.py
  1. Maybe you want to evaluate the accuracy
python eval.py