CIFAR10 Data Verification

cifar10_data_display.py

import matplotlib.pyplot as plt

class_names = ['airplane', 'automobile', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck']

dataset = tf.data.TFRecordDataset([GCS_PATH_FOR_DATA + 'train.tfrecord'])
plt.figure(figsize=(10, 10))
for i, example in enumerate(dataset.take(16)):
  data = tf.train.Example()
  data.ParseFromString(example.numpy())
  image = tf.constant(data.features.feature['image'].float_list.value, shape=[32, 32, 3])
  label = data.features.feature['label'].int64_list.value[0]
  plt.subplot(4, 4, i+1)
  plt.xticks([])
  plt.yticks([])
  plt.grid(False)
  plt.imshow(image.numpy())
  plt.xlabel(class_names[label])
plt.show()