Чтение файлов изображений

У меня есть каталог, содержащий 4 папки (1,2,3,4). В каждой папке есть изображения в формате jpg. Я использовал код ниже, чтобы прочитать изображения. Проблема в том, что все изображения имеют разную форму. Итак, теперь у меня есть список изображений, каждый из которых имеет различную форму.

1) Есть ли лучший способ прочитать файлы img из каталога? (может быть назначен непосредственно на массив NumPy)

2) Как я могу изменить размеры изображений, чтобы они все имели одинаковую форму?

Спасибо!

import imageio
import os.path

images = []

for folder in os.listdir('images'):
    for filename in os.listdir('images/'+folder):
        if filename.endswith(".jpg"): 
            img = imageio.imread('images/'+folder+'/'+filename)
            img.reshape((1,img.flatten().shape[0])).shape
            images.append(img)

0 ответов

Другие вопросы по тегам