Skip to content

Files

Latest commit

ec2108c · Jan 2, 2019

History

History
45 lines (35 loc) · 1.04 KB

README.md

File metadata and controls

45 lines (35 loc) · 1.04 KB

DarkNet Classifier LeNet MNIST

LeNet Training & Inference for MNIST dataset on DarkNet.

Requirements

  • DarkNet built
  • pip install opencv-python
  • pip install numpy

Download & Convert MNIST data

python download_and_convert_mnist.py

Set it up for DarkNet

  1. Copy the contents of cfg folder in this repo to darknet/cfg
  2. Move mnist_images folder to darknet/data/ folder
  3. Then,
cd darknet/data/mnist_images
find `pwd`/train -name \*.png > train.list
find `pwd`/test -name \*.png > test.list
cd ../..

Train

./darknet classifier train cfg/mnist.data cfg/mnist_lenet.cfg

Validate

./darknet classifier valid cfg/mnist.data cfg/mnist_lenet.cfg backup/mnist_lenet.backup

Predict

./darknet classifier predict cfg/mnist.data cfg/mnist_lenet.cfg backup/mnist_lenet.weights data/mnist_images/test/<image_name>.png

Credits

https://pjreddie.com/darknet/train-cifar/
https://github.com/ashitani/darknet_mnist