for image, label in val_ds.take(10):
print("Label: ", label.numpy())
Label:  [1 1 1 1 1 1 1 1 1 2 2 1 1 1 1 1 1 2 1 1 2 1 2 1 1 1 2 1 1 1 1 2]
Label:  [2 2 2 1 2 2 1 2 2 1 2 1 1 2 1 2 2 2 2 2 1 2 1 2 1 2 1 1 2 1 1 1]
Label:  [1 1 2 1 2 2 1 1 1 2 2 2 1 1 1 1 2 2 1 2 2 2 1 2 1 2 2 2 2 1 1 1]
Label:  [1 1 1 1 1 1 2 2 2 2 1 2 2 1 2 1 1 1 1 1 2 2 1 2 1 1 1 2 2 2 2 1]
Label:  [1 2 2 1 1 2 1 1 1 2 1 1 1 2 1 1 2 1 2 1 1 2 1 1 1 1 2 1 1 1 2 1]
Label:  [2 2 1 1 1 1 1 2 1 2 1 2 1 2 2 2 2 2 1 1 1 1 1 1 1 1 2 1 1 1 1 1]
Label:  [1 2 2 1 1 2 1 2 2 1 2 1 1 2 1 2 2 1 1 1 2 1 2 1 1 2 2 2 1 1 1 2]