text_plain_output_1.png
import os

print('Number of images in the training set:', len(os.listdir('../input/train_images')))
print('Number of images in the test set:', len(os.listdir('../input/test_images')))