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)