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，您可以添加参数“类”。这是一个可选的类子目录列表（例如 ['dogs', 'cats']）。默认值：无。

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