Example of the data augmentation and views from the plot

data_augmentation.py

data_augmentation = keras.Sequential(
    [
        layers.RandomRotation(0.3),
        layers.RandomZoom(0.2),
        layers.RandomFlip("horizontal"),
    ]
)

plt.figure(figsize=(10, 10))
for images, _ in test_dataset.take(1):
    for i in range(9):
        augmented_images = data_augmentation(images, training=True)
        ax = plt.subplot(3, 3, i + 1)
        plt.imshow(augmented_images[0].numpy().astype("uint8"))
        plt.axis("off")