Question

我已经实现了这个简单的数据增强层，基本上它将图像旋转特定的角度（我知道它可以通过 ImageDataGenerator 完成，但这只是为了解释问题）。

    class RandomRotation(tf.keras.layers.Layer):

        def __init__(self, rotation_range=None, **kwargs):
            super(RandomRotation, self).__init__(**kwargs)    
 

        def call(self, images, training=None, **kwargs):

            batch_size = tf.shape(images)[0]

            if training is None:
               training = K.learning_phase()

            if not training:
               return images

            angles = np.random.uniform(-0.5, 0.5, batch_size)
            images = tfa.image.rotate(images, angles)

我的模型，包括这一层，然后通过 fit 方法使用 ImageDataGenerator 进行训练，以自动获取生成器。我收到此错误是因为批大小的值为无。

<块引用>

TypeError: 预期序列对象 len >= 0 或单个整数

Answer 1

tf.config.run_functions_eagerly(True)

因为脚本的第一行解决了这个问题。

处理自定义数据增强层中的批量大小 - tensorflow

1 个答案: