file mnist

examples/create_mnist.sh

@@ -0,0 +1,14 @@
+#!/usr/bin/env sh
+# This script converts the mnist data into leveldb format.
+
+cd ../data
+
+echo "Creating leveldb..."
+
+rm -rf mnist-train-leveldb
+rm -rf mnist-test-leveldb
+
+../build/examples/convert_mnist_data.bin train-images-idx3-ubyte train-labels-idx1-ubyte mnist-train-leveldb
+../build/examples/convert_mnist_data.bin t10k-images-idx3-ubyte t10k-labels-idx1-ubyte mnist-test-leveldb
+
+echo "Done."

examples/get_mnist.sh

@@ -0,0 +1,23 @@
+#!/usr/bin/env sh
+# This scripts downloads the mnist data and unzips it.
+
+echo "Downloading data..."
+
+cd ../data
+
+wget -q http://yann.lecun.com/exdb/mnist/train-images-idx3-ubyte.gz
+wget -q http://yann.lecun.com/exdb/mnist/train-labels-idx1-ubyte.gz
+wget -q http://yann.lecun.com/exdb/mnist/t10k-images-idx3-ubyte.gz
+wget -q http://yann.lecun.com/exdb/mnist/t10k-labels-idx1-ubyte.gz
+
+echo "Unzipping..."
+
+gunzip train-images-idx3-ubyte.gz
+gunzip train-labels-idx1-ubyte.gz
+gunzip t10k-images-idx3-ubyte.gz
+gunzip t10k-labels-idx1-ubyte.gz
+
+# Creation is split out because leveldb sometimes causes segfault
+# and needs to be re-created.
+
+echo "Done."

data/lenet.prototxt → examples/lenet.prototxt

@@ -3,7 +3,7 @@ layers {
   layer {
     name: "mnist"
     type: "data"
-    source: "mnist-train-leveldb"
+    source: "../data/mnist-train-leveldb"
     batchsize: 64
     scale: 0.00390625
   }

data/lenet_test.prototxt → examples/lenet_test.prototxt

@@ -3,7 +3,7 @@ layers {
   layer {
     name: "mnist"
     type: "data"
-    source: "mnist-test-leveldb"
+    source: "../data/mnist-test-leveldb"
     batchsize: 100
     scale: 0.00390625
   }

examples/train_mnist.sh

@@ -0,0 +1,3 @@
+#!/usr/bin/env sh
+
+GLOG_logtostderr=1 ../build/tools/train_net.bin ../models/lenet_solver.prototxt