summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorEvan Shelhamer <shelhamer@imaginarynumber.net>2014-02-24 22:22:30 -0800
committerEvan Shelhamer <shelhamer@imaginarynumber.net>2014-02-26 12:37:44 -0800
commit650b7b16dbbf0ee7b3a7362fce0821315d608d54 (patch)
tree8ae6b083e37dbbedb4c50b32abd40183675ce82c
parentb09adf7858d7cc401d7e42ef369efe74a45ecc0c (diff)
downloadcaffeonacl-650b7b16dbbf0ee7b3a7362fce0821315d608d54.tar.gz
caffeonacl-650b7b16dbbf0ee7b3a7362fce0821315d608d54.tar.bz2
caffeonacl-650b7b16dbbf0ee7b3a7362fce0821315d608d54.zip
file mnist
-rwxr-xr-xdata/train_mnist.sh3
-rwxr-xr-xexamples/create_mnist.sh14
-rwxr-xr-xexamples/get_mnist.sh23
-rw-r--r--examples/lenet.prototxt (renamed from data/lenet.prototxt)2
-rw-r--r--examples/lenet_solver.prototxt (renamed from data/lenet_solver.prototxt)0
-rw-r--r--examples/lenet_test.prototxt (renamed from data/lenet_test.prototxt)2
-rwxr-xr-xexamples/train_mnist.sh3
7 files changed, 42 insertions, 5 deletions
diff --git a/data/train_mnist.sh b/data/train_mnist.sh
deleted file mode 100755
index f110e0bc..00000000
--- 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
index 00000000..12707a4d
--- /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
index 00000000..e50a01a6
--- /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
index f5877ae4..e1049f73 100644
--- 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
index d0edc0f0..d0edc0f0 100644
--- a/data/lenet_solver.prototxt
+++ b/examples/lenet_solver.prototxt
diff --git a/data/lenet_test.prototxt b/examples/lenet_test.prototxt
index 676a2a6a..38f1a5e9 100644
--- 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
index 00000000..4551a4e2
--- /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