0

我已经实现了这个简单的数据增强层，基本上它将图像旋转一个特定的角度（我知道它可以通过 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)

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

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

4

1 回答 1

0 
tf.config.run_functions_eagerly(True)

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

于 2021-07-28T10:18:04.093 回答