projects
/
platform
/
upstream
/
caffeonacl.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
|
inline
| side by side (parent:
f171796
)
mnist network generation
author
Yangqing Jia
<jiayq84@gmail.com>
Mon, 30 Sep 2013 21:10:04 +0000
(14:10 -0700)
committer
Yangqing Jia
<jiayq84@gmail.com>
Mon, 30 Sep 2013 21:10:04 +0000
(14:10 -0700)
src/caffe/test/data/mnist.py
patch
|
blob
|
history
diff --git
a/src/caffe/test/data/mnist.py
b/src/caffe/test/data/mnist.py
index
d478dc7
..
02ae7f5
100644
(file)
--- a/
src/caffe/test/data/mnist.py
+++ b/
src/caffe/test/data/mnist.py
@@
-17,4
+17,14
@@
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())
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
+del db
+
+mnist = core_layers.MNISTDataLayer(
+ rootfolder=MNIST_ROOT, name='mnist', is_training = False)
+db = leveldb.LevelDB('mnist-test-leveldb')
+
+for i in range(10000):
+ 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