将图像加载到Jupyter Notebook

Question

从文件夹加载图像时出现错误。我已经加载了以下内容：

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/'

当我运行下面的代码块

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

我收到一条错误消息，提示“不是目录”。

Answer 1

一个简单的方法是使用glob模块，例如：

import glob

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

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