from __future__ import absolute_importfrom keras.datasets.cifar import load_batchfrom keras.utils.data_utils import get_fileimport numpy as npimport osdef load_data():dirname = "cifar-10-batches-py"path = get_file(dirname, origin=origin, untar=True)nb_train_samples = 50000X_train = np.zeros((nb_train_samples, 3, 32, 32), dtype="uint8")y_train = np.zeros((nb_train_samples,), dtype="uint8")for i in range(1, 6):fpath = os.path.join(path, 'data_batch_' + str(i))bpath = os.path.join(path, 'batches.meta')data, labels = load_batch(fpath)label_names = load_batch(bpath)return data, labels, label_namesdata, labels, label_names = load_data()print(labels[10], label_names[7])
---------------------------------------------------------------------------KeyError Traceback (most recent call last)<ipython-input-15-cf7a7f68853d> in <module>() 24 return data, labels, label_names 25 ---> 26 data, labels, label_names = load_data() 27 28 print(labels[10], label_names[7]) <ipython-input-15-cf7a7f68853d> in load_data() 20 bpath = os.path.join(path, 'batches.meta') 21 data, labels = load_batch(fpath) ---> 22 label_names = load_batch(bpath) 23 24 return data, labels, label_names /home/robotics/anaconda3/lib/python3.5/site-packages/keras/datasets/cifar.py in load_batch(fpath, label_key) 17 d[k.decode("utf8")] = v 18 f.close() ---> 19 data = d["data"] 20 labels = d[label_key] 21 KeyError: 'data'
I'm trying to display how my model does via displaying the target image and my model prediction with meaningful names.