diff --git a/read_cifar.py b/read_cifar.py index 3d2709e6111a9048dd7cf2534722178d8295dd97..9e2a47d51c48a0cbbe114d818d706fcd91c78d58 100644 --- a/read_cifar.py +++ b/read_cifar.py @@ -14,7 +14,7 @@ def read_cifar_batch(path): 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") + data, labels = read_cifar_batch("./data/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]))