Load in images to jupyter notebook

Question

I am getting an error loading in images from folders. I have loaded in the following:

base_path = '/Users/rayday/Desktop/COMP309_2020_Project/'
train_dir = '/Users/rayday/Desktop/COMP309_2020_Project/Train_data/'
test_dir ='/Users/rayday/Desktop/COMP309_2020_Project/ProjectTemplate_python3.7/data/test/'

When I run the below chunk of code

print('Training Data')
for direc in os.listdir(train_dir):
    print(direc, ':',len(os.listdir(train_dir+direc)))

I get an error saying 'Not a directory'.

Answer 1

A easy way to do this is with the glob module, for example:

import glob

image_paths = glob.glob('images/*.jpg')

for image_path in image_paths:
    # Load image
    original_image = Image.open(image_path)