在 Keras 中为 flow_from_directory 使用多个目录

Tha*_*yen 6 python keras tensorflow

我的场景是我们有多个拥有自己数据的对等点,位于不同的目录中,具有相同的子目录结构。我想使用这些数据训练模型,但是如果我将它们全部复制到一个文件夹中,我将无法跟踪哪些数据来自谁(偶尔也会创建新数据,因此不适合保留复制文件每次)我的数据现在存储如下:

-user01
-user02
-user03
...
Run Code Online (Sandbox Code Playgroud)

(它们都有相似的子目录结构)

我已经搜索了解决方案,但我只在这里这里找到了多输入案例,它们将多个输入连接成 1 个单个并行输入,这不是我的情况。

我知道flow_from_directory()一次只能由 1 个目录提供,那么如何构建一个可以一次由多个目录提供的自定义目录?

如果我的问题是低质量的,请提供有关如何改进它的建议,我也在 keras 的 github 上进行了搜索,但没有找到任何我可以适应的内容。

谢谢你。

use*_*622 2

Keras ImageDataGenerator flow_from_directory方法有一个follow_links参数。

也许您可以创建一个目录,其中填充了所有其他目录中文件的符号链接。

此堆栈问题讨论使用 Keras ImageDataGenerator 的符号链接:Understanding 'follow_links' argument in Keras's ImageDataGenerator?