CIFAR-10数据库的提取

Question

我试图提取CIFAR 10数据库使用此代码，但我得到以下错误，请帮助我删除此错误。

还有其他方法来提取这个数据库吗。我试图解决这个错误，但我做不到。请帮帮我。

import numpy as np
import matplotlib.pyplot as plt
import pickle
import os
def unpickle(file):
    if os.path.getsize(file) > 0: 
    # load files from dict
        with open(file, 'rb') as f:
            data = pickle.load(f, encoding='bytes')
        return data


def load_cifar10_data(data_dir):
    #Return train_data, train_labels, test_data, test_labels
    train_data = None
    train_labels = []

    for i in range(1, 6):
        data_dic = unpickle(data_dir + "/data_batch_{}".format(i))
    if i == 1:
        train_data = data_dic['data']
    else:
        train_data = np.vstack((train_data, data_dic['data']))

    train_labels += data_dic['labels']

    test_data_dic = unpickle(data_dir + "/test_batch")
    test_data = test_data_dic['data']
    test_labels = test_data_dic['labels']

    train_data = train_data.reshape((len(train_data), 3, 32, 32))
    train_data = np.rollaxis(train_data, 1, 4)
    train_labels = np.array(train_labels)

    test_data = test_data.reshape((len(test_data), 3, 32, 32))
    test_data = np.rollaxis(test_data, 1, 4)
    test_labels = np.array(test_labels)

    return train_data, train_labels, test_data, test_labels

data_dir = 'D:\Database\cifar-10-batches-py'

train_data, train_labels, test_data, test_labels = load_cifar10_data(data_dir)

print(train_data.shape)
print(train_labels.shape)

print(test_data.shape)
print(test_labels.shape)

# In order to check where the data shows an image correctly
plt.imshow(train_data[2])
plt.show()

错误：

load_cifar10_data train_data = np.vstack((train_data，data_dic‘data’)中的"D:/Program /Spyder文件/深度学习/untled6.py“文件，第37行) KeyError：“数据”

Answer 1

很抱歉这么晚才回答，但是使用data_dic[b'data']和data_dic[b'labels']，因为它是字节格式的。