diff options
author | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2014-02-24 22:22:30 -0800 |
---|---|---|
committer | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2014-02-26 12:37:44 -0800 |
commit | 650b7b16dbbf0ee7b3a7362fce0821315d608d54 (patch) | |
tree | 8ae6b083e37dbbedb4c50b32abd40183675ce82c | |
parent | b09adf7858d7cc401d7e42ef369efe74a45ecc0c (diff) | |
download | caffeonacl-650b7b16dbbf0ee7b3a7362fce0821315d608d54.tar.gz caffeonacl-650b7b16dbbf0ee7b3a7362fce0821315d608d54.tar.bz2 caffeonacl-650b7b16dbbf0ee7b3a7362fce0821315d608d54.zip |
file mnist
-rwxr-xr-x | data/train_mnist.sh | 3 | ||||
-rwxr-xr-x | examples/create_mnist.sh | 14 | ||||
-rwxr-xr-x | examples/get_mnist.sh | 23 | ||||
-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-x | examples/train_mnist.sh | 3 |
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 |