如何使用keras flow_from_directory shuffled方法检索图片的文件名？

Question

如果我不对我的文件进行混洗，我可以用generator.filenames获得文件名。但是，当生成器对图像进行混洗时，文件名不会被混洗，因此我不知道如何取回文件名。

Answer 1

在内部，DirectoryIterator将遍历一个index_arrray，当您请求它时，它会被洗牌。

您只需要使用批处理的当前索引对文件名数组进行索引：

it = ImageDataGenerator().flow_from_directory(shuffle=True,...)
for img in it:
    idx = (it.batch_index - 1) * it.batch_size
    fnames = [it.filenames[it.index_array[i]] for i in range(idx, idx + it.batch_size)]

Answer 2

我认为这里唯一的选择就是不对文件进行混洗。我自己一直在想，这是我在文档中唯一能找到的东西。看起来很奇怪也不正确。