Произвольное преобразование всего в ImageDataGenerator за один раз в tf.keras?

Question

Я использую tennsflow.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)))

Это лучший способ изменить все сразу?

Answer 1

Вы можете просто позвонить fit_generator вместо fit и next.Таким образом, вам не нужно будет перебирать все ваши данные.Для получения дополнительной информации о fit_generator обратитесь к keras help