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獲得。