ImageDataGenerator keras - save_to_dir

经过这里的一些实验，你可以如何做到这一点。下面的代码将提供原始文件名和相关标签。标签是与类对应的整数类索引。即 filenames[0] 具有标签 labels[0] 等。 class_indices 属性是一个字典，包含从类名到类索引的映射。例如，如果您有 2 个类猫和狗，则字典将为 {'cats': 0, 'dogs': 1}。现在事实证明，如果您使用 shuffle=False 运行 image_data_generator，生成器会使用随机文件名保存增强文件，但是第二个下划线之前的第一个数字似乎是处理文件的顺序。所以 _0_random number 是第一个处理的文件。所以它在下面代码中的关联标签将是 labels[0] 等。因此，您可以通过这种方式确定保存的图像属于哪个类。现在您可能想使用 shuffle=True 进行训练。现在生成器打乱了图像，但是第二个下划线之前的数字仍然代表原始图像顺序。因此，如果它被标记为 _10_random 数，则此图像标签为 label[10]。

data_gen=ImageDataGenerator.flow(
    x,
    y,
    batch_size=32,
    shuffle=True,
    sample_weight=None,
    seed=None,
    save_to_dir= "path_to_dir",
    save_prefix="",
    save_format="png",
    subset=None,
file_names= data_gen.filenames
labels= data_gen.labels
label_dict=data_gen.class_indices