将CIFAR-10数据导入python

根据官网的说明,需要使用python3下的pickle

  • 首先把官网的python版本数据下载保存到本地
  • 进行解压,会得到如下


    cifar.png
  • 下面就可以进行数据的导入了
file1 = 'D:/02 document/cs231n/data/cifar-10-batches-py/data_batch_1'#文件的路径
def unpickle(file):#CIFAR-10官方给出的使用方法
    import pickle
    with open(file, 'rb') as fo:
        dict = pickle.load(fo, encoding='iso-8859-1')
    return dict


dict_train_batch1 = unpickle(file1)  # 将data_batch文件读入到数据结构(字典)中
data_train_batch1 = dict_train_batch1.get('data')  # 字典中取data
labels = dict_train_batch1.get('labels')  # 字典中取labels
#可使用print来观察导入的数据
print(data_train_batch1)
print(labels)

你可能感兴趣的:(将CIFAR-10数据导入python)