Skip to content

Latest commit

 

History

History
20 lines (16 loc) · 467 Bytes

README.md

File metadata and controls

20 lines (16 loc) · 467 Bytes

keras-learnings

Install the requirements.txt if needed.

Activate the virtual env:

source env/bin/activate

Train like this:

python train_simple_nn.py --dataset animals --model output/simple_nn.model \
--label-bin output/simple_nn_lb.pickle --plot output/simple_nn_plot.png

Run like this:

python predict.py --image hoop2.jpg --model output/simple_nn.model \
--label-bin output/simple_nn_lb.pickle --width 32 --height 32 --flatten 1