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

Posted 2021-05-03

文件夹的结构如下：

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

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

在这种情况下，如何使用keras的ImageDataGenerator.flow_from_directory，因为这需要我在不同的子文件夹中对图像进行排序？我的训练集的容量大约为500 G，根据班级的标签构建子文件夹并不方便。如果我不能使用ImageDataGenerator.flow_from_directory，是否有其他方法可以实现此功能？

答案

Keras ImageDataGenerator类也提供flow_from_dataframe()方法，该方法可以完成此工作。

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