keras datagen.flow_from_directory类模式类别获取错误标签

Question

我有24个类的图像训练数据集，每个类有2800个图像这些类在一个名为训练的目录中相互分离

我使用以下代码使用keras加载数据：

datagen = ImageDataGenerator(rescale= 1./255
,fill_mode='nearest')
train_data = datagen.flow_from_directory('training-images',class_mode='categorical',
target_size=(32,32))

它加载图像，但图像得到错误的标签

目录结构如下：

training-images/
0/
image.jpg
.
.
1/
image.jpg
.
.
.
.
23/
image.jpg
.
.

Answer 1

flow_from_directory按字母顺序排列类，而不是按数字排列。这意味着您的课程将按0、1、11、12、13、14、15、16、17、18、19、2、20、21、22、23、24、3、4、5、6、7、8、9的顺序排列。如果我是你，我会给文件夹提供描述性名称(例如，对于CIFAR10，它们将是"飞机"、"鸟"等(，然后使用.class_indices将生成的ID映射到类名。