如何在 Keras 中为不同批次应用不同的图像生成器？

Question

从两个 .npy 文件中读取训练数据。也就是说，train_set 被视为 X，train_label 被视为 Y。因此，它不是多输入的情况。我的任务需要以不同的方式增加图像补丁。那么如何为不同的patch定义不同的Image Generator呢？虽然可能有很多补丁，但我以 3 个补丁为例：

对于补丁 1：

datagen = ImageDataGenerator(rotation_range=20)

对于补丁 2：

datagen = ImageDataGenerator(rotation_range=40)

补丁3：

datagen = ImageDataGenerator(rotation_range=60)

如何在不同的补丁上应用不同的生成器，以及如何将 model.fit(...) 或 model.fit_generator(...) 用于所描述的场景？

另外，有没有办法将图像旋转特定角度而不是范围？

谢谢！

Answer 1

我自己没有这样做，但我认为一种方法是使用第一个数据生成并使用 fit_generator 和选定的时期数传递第一组训练数据。然后，保存权重并使用第二个数据生成和第二组与 fit_generator。您还需要设置 initial_epoch 并且还需要加载权重。为了概括这个问题，您需要做的是使用第二个数据生成恢复训练。请参阅https://keras.io/getting-started/faq/#how-can-i-save-a-keras-model。