通过互联网搜索后,我无法找到如何使用以下数据将一组带有相应标签的图像保存为批处理文件我将能够检索(加载)它们为他们工作。简单地说我正在寻找一种方法来保存它们,以便使用以下代码来检索它们。
import cPickle as pickle
import numpy as np
import os
def load_CIFAR_batch(filename):
""" load single batch of cifar """
with open(filename, 'rb') as f:
datadict = pickle.load(f)
X = datadict['data']
Y = datadict['labels']
X = X.reshape(10000, 3, 32, 32).transpose(0,2,3,1).astype("float")
Y = np.array(Y)
return X, Y
e.g。 CIFAR10的数据集已经是使用cPickle的批量格式,但我不知道如何使用cPickle来保存图像。
CIFAR-10数据集链接:http://www.cs.toronto.edu/~kriz/cifar.html
我正在使用:
Ubuntu 14.04 LTS Python 2.7