Step 2 – Loading the training/test images to generate train/test set