python - 在 tf.keras 中一次随机转换 ImageDataGenerator 中的所有内容？

Question

我正在使用 tensorflow.keras.utils.Sequence 到 model.fit_generator。我正在检索数据并一次改组一批，而不是将所有内容加载到内存中。在我的__init__，我有self.datagen = ImageDataGenerator(width_shift_range=0.2, height_shift_range=0.2, zoom_range=0.2)。然后在我的__getitem__，我有：

self.datagen.fit(x_batch)
x_batch = next(self.datagen.flow(x_batch, batch_size=len(x_batch)))

这是一次改变一切的最佳方式吗？

标签： pythontensorflowkeras

Answer 1

你可以打电话fit_generator而不是fitand next。这样，您就不需要遍历所有数据。有关fit_generator查看keras 帮助的更多信息