# -- Markdown Cell --
# # Starter Kit
# 
# This starter kit contains plots of some samples from the training and test datasets. It also serves as an example of how to read the images in the dataset.

# -- Code Cell --
import matplotlib.pyplot as plt
import PIL.Image as Image

# -- Markdown Cell --
# ## Plot Some Test Samples

# -- Code Cell --
fig, ax = plt.subplots(5, 5)

fig.subplots_adjust(left=0, bottom=0, right=3, top=4, wspace=0, hspace=0)

for i in range(25):
    path = "test_data/img_" + str(i) + ".png"
    img = Image.open(path)
    ax[i // 5][i % 5].imshow(img)
    ax[i // 5][i % 5].set_axis_off()

# -- Markdown Cell --
# ## Plot Some Train Samples

# -- Code Cell --
fig, ax = plt.subplots(5, 5)

fig.subplots_adjust(left=0, bottom=0, right=3, top=4, wspace=0, hspace=0)

for i in range(25):
    path = "train_data/img_" + str(i) + ".png"
    img = Image.open(path)
    ax[i // 5][i % 5].imshow(img)
    ax[i // 5][i % 5].set_axis_off()