数据集一般有key为
dict_keys([b’batch_label’, b’labels’, b’data’, b’filenames’])
Python2版本的key没有b’’,即
dict_keys([‘batch_label’, ‘labels’, ‘data’, ‘filenames’])
这是不同版本Python之间的区别
import pickle
def load(filename):
with open(filename, 'rb') as fo:
data = pickle.load(fo, encoding='latin1')
return data
d = 'test_batch' #数据集“test_batch”
dataset_path_1 = load(d)
print(dataset_path_1.keys())
print(dataset_path_1)
# print(dataset_path_1['batch_label']) <class 'bytes'>
# print(dataset_path_1[b'labels']) <class 'list'>
# print(dataset_path_1['data'].shape) <class 'numpy.ndarray'>
# print(dataset_path_1['filenames']) <class 'list'>