Может ли Keras.load_img работать с несколькими изображениями в папке

Question

Я работаю над алгоритмом классификации изображений собак и кошек с использованием Keras. Согласно приведенному ниже коду, это работает, когда я загружаю одно изображение, но мой вопрос: может ли Keras.load_img работать с несколькими изображениями в папке? Когда я пытаюсь это сделать, он говорит, что у объекта списка нет атрибутов поиска и чтения.

from keras.preprocessing import image
test_image = image.load_img('dataset/single_prediction/cat_or_dog_1.jpg', target_size = (64, 64))
test_image = image.img_to_array(test_image)
test_image = np.expand_dims(test_image, axis = 0)
result = classifier.predict(test_image)
training_set.class_indices
if result[0][0] == 1:
prediction = 'dog'
else:
prediction = 'cat'

Answer 1

Я думаю, что есть решение, вы можете попробовать это:

images = []
for img in os.listdir(folder_path):
    img = image.load_img(img, target_size=(img_width, img_height))
    img = img.img_to_array(img)
    img = np.expand_dims(img, axis=0)
    images.append(img)
images = np.vstack(images)
classes = classifier.predict(images, batch_size=10)
print(classes)