file mnist

diff --git a/data/train_mnist.sh b/data/train_mnist.sh
deleted file mode 100755 (executable)
index f110e0b..0000000
--- a/data/train_mnist.sh
+++ /dev/null
@@ -1,3 +0,0 @@
-#!/usr/bin/env sh
-
-GLOG_logtostderr=1 ../build/examples/train_net.bin lenet_solver.prototxt

diff --git a/examples/create_mnist.sh b/examples/create_mnist.sh
new file mode 100755 (executable)
index 0000000..12707a4
--- /dev/null
+++ b/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."

diff --git a/examples/get_mnist.sh b/examples/get_mnist.sh
new file mode 100755 (executable)
index 0000000..e50a01a
--- /dev/null
+++ b/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."

diff --git a/data/lenet.prototxt b/examples/lenet.prototxt
similarity index 97%
rename from data/lenet.prototxt
rename to examples/lenet.prototxt
index f5877ae..e1049f7 100644 (file)
--- a/data/lenet.prototxt
+++ b/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
   }

diff --git a/data/lenet_solver.prototxt b/examples/lenet_solver.prototxt
similarity index 100%
rename from data/lenet_solver.prototxt
rename to examples/lenet_solver.prototxt

diff --git a/data/lenet_test.prototxt b/examples/lenet_test.prototxt
similarity index 97%
rename from data/lenet_test.prototxt
rename to examples/lenet_test.prototxt
index 676a2a6..38f1a5e 100644 (file)
--- a/data/lenet_test.prototxt
+++ b/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
   }

diff --git a/examples/train_mnist.sh b/examples/train_mnist.sh
new file mode 100755 (executable)
index 0000000..4551a4e
--- /dev/null
+++ b/examples/train_mnist.sh
@@ -0,0 +1,3 @@
+#!/usr/bin/env sh
+
+GLOG_logtostderr=1 ../build/tools/train_net.bin ../models/lenet_solver.prototxt