python - 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 按字母顺序(而不是数字)对类进行排序。这意味着您的类(class)将按 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 映射到类名称.