Create data.py

llp1992 · llp1992 · commit ba49bfc44d13 · 2015-08-28T16:41:39.000+08:00
diff --git a/DeepLearning/CNN_mnist/data.py b/DeepLearning/CNN_mnist/data.py
@@ -0,0 +1,19 @@
+# -*- coding: utf-8 -*-
+import os
+from PIL import Image
+import numpy as np
+
+def load_data():
+    print 'start loading data...'
+    data = np.empty((42000,1,28,28),dtype="float32")
+    label = np.empty((42000,),dtype="uint8")
+
+    imgs = os.listdir("./mnist")
+    num = len(imgs)
+    for i in range(num):
+        img = Image.open("./mnist/"+imgs[i])
+        arr = np.asarray(img,dtype="float32")
+        data[i,:,:,:] = arr
+        label[i] = int(imgs[i].split('.')[0])
+    return data,label
+
