Dataframe 在现有数据拆分上构建tf.keras.preprocessing.image.ImageDataGenerator

假设我们有两个带有文件名的txt文件（train split和validation split）。 现在我们想使用

tf.keras.preprocessing.image.ImageDataGenerator

。在基于文件夹结构（每个类）的随机拆分的情况下，生成器将这样构建

TRAINING_DATA_DIR = str(data_root)
datagen_kwargs = dict(rescale=1. / 255, validation_split=0.2)  # rescale and split

valid_datagen = tf.keras.preprocessing.image.ImageDataGenerator(**datagen_kwargs)
valid_generator = valid_datagen.flow_from_directory(
    TRAINING_DATA_DIR,
    subset="validation",
    shuffle=True,
    target_size=IMAGE_SHAPE)

但是，如何基于包含拆分图像名称的两个txt文件构建生成呢

我发现除了

flow\u from\u directory

函数之外，还有一个from\u dataframe函数，但它仍然没有给出问题的答案

任何想法都将受到高度赞赏

谢谢