给定从图像名称到类标签的映射，如何使用 keras ImageDataGenerator flow_from_directory？

Question

文件夹结构如下：

-dataset
   |---train_set
   |      |-- img01.jpg
   |      |-- img02.jpg
   |      |-.....
   |
   |---val_set 

我在 train_set 中没有名称为类标签的子文件夹。但是我有一个从图像名称到类标签的字典，例如 map = {'img01.jpg': 10, 'img02.jpg': 3, 'img03.jpg':12, ...}

在这种情况下如何使用 keras 的ImageDataGenerator.flow_from_directory，因为它需要我对不同子文件夹中的图像进行排序？我的训练集大小在 500 G 左右，不方便根据类的标签建立子文件夹。如果我不能使用ImageDataGenerator.flow_from_directory，有没有其他方法可以实现这个功能？

Answer 1

Keras ImageDataGenerator 类还提供了flow_from_dataframe() 方法，应该可以完成这项工作。

查看此tutorial（第二部分）作为您的案例示例。