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

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

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