A LeNet5 implementation in Python3 using Keras
Вы не можете выбрать более 25 тем Темы должны начинаться с буквы или цифры, могут содержать дефисы(-) и должны содержать не более 35 символов.
pry a04b1e4e4b added weights file 1 год назад
.gitignore Add inital commit for LeNet5 1 год назад
LICENSE Initial commit 1 год назад
README.md improved docs 1 год назад
lenet5.py Add inital commit for LeNet5 1 год назад
lenet5_weights.hdf5 added weights file 1 год назад
main.py improved docs 1 год назад
requirements.txt Add inital commit for LeNet5 1 год назад

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.