將圖像加載到 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)