A LeNet5 implementation in Python3 using Keras
Nelze vybrat více než 25 témat Téma musí začínat písmenem nebo číslem, může obsahovat pomlčky („-“) a může být dlouhé až 35 znaků.
pry a04b1e4e4b added weights file před 1 rokem
.gitignore Add inital commit for LeNet5 před 1 rokem
LICENSE Initial commit před 1 rokem
README.md improved docs před 1 rokem
lenet5.py Add inital commit for LeNet5 před 1 rokem
lenet5_weights.hdf5 added weights file před 1 rokem
main.py improved docs před 1 rokem
requirements.txt Add inital commit for LeNet5 před 1 rokem

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.