Keras：使用文件名而不是目录的flow_from_directory（）或flow（）

Question

我正在从目录中将图像导入Keras。代码看起来像这样：

gen = datagen.flow_from_directory (
    DIRECTORY,
    target_size = (IMAGE_WIDTH, IMAGE_HEIGHT),
    batch_size = BATCH_SIZE,
    class_mode = "binary"
)

但是，我的数据不是由文件夹组织，而是由文件名组织。如果我无法更改此结构，是否有任何方法可以在通过正则表达式分隔时流动（例如，以“.jpg”与“.png”结尾，或以“numeral_7”与“numeral_8”开头）？

Answer 1

根据official documentation，您可以添加参数“classes”。这是类子目录的可选列表（例如['dogs'，'cats']）。默认值：无。

如果没有提供，将从目录下的子目录名称/结构自动推断类列表，其中每个子目录将被视为不同的类（以及将映射到标签索引的类的顺序，将是字母数字）。包含从类名到类索引的映射的字典可以通过属性class_indices获得。