A LeNet5 implementation in Python3 using Keras
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
pry 5c520e4437 improved docs 2 months ago
.gitignore Add inital commit for LeNet5 2 months ago
LICENSE Initial commit 3 months ago
README.md improved docs 2 months ago
lenet5.py Add inital commit for LeNet5 2 months ago
main.py improved docs 2 months ago
requirements.txt Add inital commit for LeNet5 2 months ago

README.md

lenet5

A LeNet5 implementation in Python3 using Keras

Set up

  1. Clone the repo git clone https://git.prydt.xyz/pry/lenet5 cd lenet5

  2. Set up a python virtual environment with virtualenv in the repo virtualenv .

  3. Install all the necessary packages ./bin/pip install -r requirements.txt

  4. Just run main.py with the python installed with virtualenv

Example (train a model and save it to output.hdf5): ./bin/python main.py --save-model 1 --weights output.hdf5

Acknowledgements

This codebase is roughly following this good article.