tensorflow2.x - 属性错误 : module 'tensorflow_core.keras.layers.experimental.preprocessing' has no attribute 'RandomFlip'

Question

我使用 Tensorflow 2.1.0
在这段代码中

data_augmentation = tf.keras.Sequential([
    tf.keras.layers.experimental.preprocessing.RandomFlip('horizontal'),
    tf.keras.layers.experimental.preprocessing.RandomRotation(0.3)
]) 

我发现这个错误:

AttributeError: module 'tensorflow_core.keras.layers.experimental.preprocessing' has no attribute 'RandomFlip'

那么如何在不更改 tensorflow 版本的情况下更改它

Answer 1

要按预期运行您的代码，首先必须将 Tensorflow 升级到最新版本

! pip install tensorflow --upgrade

如果您正在 TF 2.1.0 中寻找解决方案，那么有两个选项可用
第一个解决方案: tf.image.random_flip_left_right (水平翻转)

tf.image.random_flip_left_right(
    image, seed=None)

第二种解决方案: tf.keras.preprocessing.image.ImageDataGenerator

tf.keras.preprocessing.image.ImageDataGenerator(
   rotation_range=30, horizontal_flip=True)