Keras-使用预定义的训练/验证拆分

Question

我正在使用Tensorflow / Keras。我有两个文本文件（train_{modality_name}.txt和val_{modality_name}.txt）。它们包含我要用于处理图像的拆分。

这些文件的格式如下：

example_0_path category_id
example_1_path category_id
...
example_N_path category_id

我的文件夹结构是这样的：

/labels
    train_X.txt
    val_X.txt
/data
    /modality_1
    ...
    /modality_M
 (e.g. data/sketch/abbey/id)

如何使用文件？

Answer 1

'flow_from_dataframe'完成了这项工作，此外还需要用熊猫预处理txt。本教程非常有帮助：https://medium.com/@vijayabhaskar96/tutorial-on-keras-imagedatagenerator-with-flow-from-dataframe-8bd5776e45c1

仍然无法匹配数组的目标大小（标签似乎格式错误）