提交 7d637375 编写于 作者: Y yangjiahua

fix if window system do not support slash path

上级 7b9ee43d
......@@ -8,9 +8,10 @@ import numpy as np
import os
import pickle
data_dir = '../../Data/cifar-10-batches-py/'
train_o_dir = '../../Data/cifar-10-png/raw_train/'
test_o_dir = '../../Data/cifar-10-png/raw_test/'
data_dir = os.path.join("..", "..", "Data", "cifar-10-batches-py")
train_o_dir = os.path.join("..", "..", "Data", "cifar-10-png", "raw_train")
test_o_dir = os.path.join("..", "..", "Data", "cifar-10-png", "raw_test")
Train = False # 不解压训练集,仅解压测试集
......@@ -29,7 +30,7 @@ def my_mkdir(my_dir):
if __name__ == '__main__':
if Train:
for j in range(1, 6):
data_path = data_dir + "data_batch_" + str(j) # data_batch_12345
data_path = os.path.join(data_dir, "data_batch_" + str(j)) # data_batch_12345
train_data = unpickle(data_path)
print(data_path + " is loading...")
......@@ -49,7 +50,7 @@ if __name__ == '__main__':
print("test_batch is loading...")
# 生成测试集图片
test_data_path = data_dir + "test_batch"
test_data_path = os.path.join(data_dir, "test_batch")
test_data = unpickle(test_data_path)
for i in range(0, 10000):
img = np.reshape(test_data[b'data'][i], (3, 32, 32))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册