The classes of the images in the datasets are cat and non-cat. These labels of each sample are stored in train_set_y_orig, validation_y and test_set_y, where 1 means 'cat' and 0 means 'non-cat'. Let us see how to get this info Pythonically:
The following code gets the classes from the data set. We will be printing the classes, how the labels of the classes are stored in the data set. Then, we will be plotting the image.
Copy-paste the following code.
classes = np.array(test_dataset["list_classes"][:])
print("Classes are: ",classes)
print("Groundtruth stored as: ",train_set_y_orig[2])
print(classes[train_set_y_orig[2]].decode('utf-8'))
plt.imshow(train_set_x_orig[2])
Loading comments...