tensorflow - 理解 tf/keras 源代码（用于 ImageDataGenerator）

Question

我想调整 ImageDataGenerator 类，以便让我合并我自己的自定义增强功能。我无法将我的函数设置为 ImageDataGenerators kwargs 中的 preprocessing_function，因为我想在其他标准增强之前先应用此函数。

查看 ImageDataGenerator 的源代码，https://github.com/tensorflow/tensorflow/blob/v2.4.1/tensorflow/python/keras/preprocessing/image.py#L581-L1100，我不明白在哪里代码将标准扩充应用于数据。

例如，当我打电话时

image_datagen = tf.keras.preprocessing.image.ImageDataGenerator(rotation_range=30)
image_generator = image_datagen.flow_from_directory('./mydir',)

然后 mydir 中的图像将应用随机旋转功能。但是这在源代码中发生在哪里？

可能在源代码的末尾，https://github.com/tensorflow/tensorflow/blob/85c8b2a817f95a3e979ecd1ed95bff1dc1335cff/tensorflow/python/keras/preprocessing/image.py#L1103？

keras_export('keras.preprocessing.image.random_rotation')(random_rotation)

我不明白这行代码在做什么？

标签： tensorflowkeras