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

Question

我正在将tensorflow.keras.utils.Sequence用于model.fit_generator。 我正在检索数据并一次混洗一批，而不是将所有内容都加载到ram中。 在我的__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)))

这是一次转换所有内容的最佳方法吗？

Answer 1

您可以直接致电fit_generator而不是fit和next。这样，您就不需要遍历所有数据。有关fit_generator的更多信息，请查看keras help