python - 如何在 Tensorflow Keras 中标准化我的图像数据

Question

如前所述，我正在尝试在训练我的模型之前规范化我的数据集。我之前使用 tf.keras.preprocessing.image.ImageDataGenerator 来执行此操作。

        train_data = tf.cast(train_data, tf.float32)
        train_gen = ImageDataGenerator(
            featurewise_center=True,
            featurewise_std_normalization=True
        )
        train_gen.fit(train_data)
        train_generator = train_gen.flow(train_data, train_labels,
                                         batch_size=batch_size,
                                         shuffle=True)
        model.fit(train_generator, epochs=base_epochs)

但是，我不得不放弃它，因为我使用自定义层实现了一个复杂的损失函数。因此，数据和标签需要作为输入分别发送到模型。 Tensorflow Keras 是否提供了任何其他功能来规范化我的样本？

Answer 1

    def standardize(image_data):
        image_data -= np.mean(image_data, axis=0)
        image_data /= np.std(image_data, axis=0)
        return image_data

这是解决问题的简单方法。自己预处理数据。