0

我在我的训练管道上使用图像增强，但旋转会影响我的缩放。我将每个图像缩放到 (0,1) 并且我的目标设备也会这样做。所以，我想知道：

  • ImageDataGenerators 是否在本地扩展，但使用后处理器而不是预处理器
  • ImageDataGenerators 是否以某种方式相互传输
  • 是否可以拦截 model.fit(aug, ... ) 中的流，以便在将其输入系统之前对其进行规范化

还有其他方法可以实现我的意图吗？

相关代码：

aug = tf.keras.preprocessing.image.ImageDataGenerator(
   rotation_range=5,
   width_shift_range=0.1,
   height_shift_range=0.1,
   horizontal_flip=True,
   fill_mode ='nearest')

aug.fit(trainX,seed = SEED)

# ...

model2.fit(
    aug.flow(trainX, trainY, batch_size=BS),
    #trainX, trainY, batch_size=BS,
    validation_data=(testX, testY),
    steps_per_epoch=len(trainX) // BS,
    callbacks=[tensorboard_callback2,tf.keras.callbacks.LearningRateScheduler(lr_time_based_decay, verbose=1),
            stop_early
            ],
    epochs=EPOCHS)
4

0 回答 0