在keras中使用predict_generator()时如何获取关联的图像名称

Question

test_batches = ImageDataGenerator(
    preprocessing_function=preprocess_input
).flow_from_directory(test_path,target_size=(224,224),batch_size=1,class_mode=None,shuffle = "false")
prediction = model.predict_generator(test_batches, steps=1, verbose=1)
np.argmax(prediction)

因此，我在这里使用step_size=1 和steps=1 来测试一张图像。每当我运行这个时，我都会得到不同的预测，这意味着它不会每次都选择相同的图像。如何查看图片名称？

编辑：这是解释我面临的问题的另一种尝试：

test_batches = ImageDataGenerator(
    preprocessing_function=preprocess_input
).flow_from_directory(test_path,target_size=(224,224),batch_size=2,class_mode=None,shuffle = "false")
prediction = model.predict_generator(test_batches, steps=1, verbose=2)

预测变量有两个预测概率数组。我如何知道这些预测适用于哪些图像？

Answer 1

• 如果您的目标是熟悉 keras ImageDataGenerators：

如果您希望生成器始终返回相同的图像（为了可重复性）：

from keras.preprocessing.image import ImageDataGenerator
import numpy as np

data_dir = 'path/to/image/directory'  # path to the directory where the images are stored
index = 0  # select a number here

ig = ImageDataGenerator()
gen = ig.flow_from_directory(data_dir, batch_size=1)  # if you want batch_size > 1 you need to
                                                      # add as many indices as your batch_size.
image, label = gen._get_batches_of_transformed_samples(np.array([index]))
image_name = gen.filenames[index]
# do whatever you want with your image and label

如果您希望生成器始终返回随机图像但知道它是哪一个，我建议执行以下操作：

index = next(gen.index_generator)
image, label = gen._get_batches_of_transformed_samples(index)
image_name = gen.filenames[index]

• 然而，如果您想了解如何predict_generator工作，那么这些方法都无法帮助您。我唯一能想到的就是编辑DirectoryIterator 代码。

例如，您可以添加一行来打印您传递的图像的名称。我建议在第 1434 行之后添加以下语句：

print(fname)