Update read_cifar.py

read_cifar.py

@@ -12,8 +12,18 @@ def read_cifar_batch(path):
     labels = batch[b'labels']
     return np.float32(data), np.int64(labels)
 
+def read_cifar(folder_path):
+    data, labels = read_cifar_batch("./data/cifar-10-python.tar/cifar-10-batches-py~/cifar-10-batches-py/test_batch")
+    for i in range(1,5):
+        data = np.concatenate((data, read_cifar_batch(folder_path + "/data_batch_" + str(i))[0]))
+        labels = np.concatenate((labels, read_cifar_batch(folder_path + "/data_batch_" + str(i))[1]))
+    return data, labels
+
+
+
 
 if __name__ == "__main__":
-    v1, v2 = read_cifar_batch("./data/cifar-10-batches-py/cifar-10-batches-py~/cifar-10-batches-py/data_batch_1")
-    print(v1)
-    print(v2)
+    data, labels = read_cifar("./data/cifar-10-python.tar/cifar-10-batches-py~/cifar-10-batches-py")
+    print(data)
+    print(labels)
+