From: Yangqing Jia <jiayq84@gmail.com>
Date: Mon, 30 Sep 2013 16:26:22 +0000 (-0700)
Subject: mnist leveldb data generation
X-Git-Url: https://git.ti.com/gitweb?p=jacinto-ai%2Fcaffe-jacinto.git;a=commitdiff_plain;h=6bd78d59f22bef0ac15a714f31f215694318decd;ds=sidebyside

mnist leveldb data generation
---

diff --git a/src/caffe/test/data/mnist.py b/src/caffe/test/data/mnist.py
new file mode 100644
index 00000000..d478dc79
--- /dev/null
+++ b/src/caffe/test/data/mnist.py
@@ -0,0 +1,20 @@
+"""This script generates the mnist train and test leveldbs used in the
+test.
+"""
+from caffe.pyutil import convert
+from decaf.layers import core_layers
+import numpy as np
+import leveldb
+
+# the folder that has the MNIST data
+MNIST_ROOT = 'mnist'
+
+mnist = core_layers.MNISTDataLayer(
+    rootfolder=MNIST_ROOT, name='mnist', is_training = True)
+db = leveldb.LevelDB('mnist-train-leveldb')
+
+for i in range(60000):
+  datum = convert.array_to_datum((mnist._data[i] * 255).reshape(1,28,28).astype(np.uint8))
+  datum.label = mnist._label[i]
+  db.Put('%d' % (i), datum.SerializeToString())
+del db
\ No newline at end of file