修改下面代码使其具体生成器的卷积lstm更换为卷积RNN def build_generator(self): def res_net(inputs, filters): x = inputs net = conv2d(x, filters // 2, (1, 1), 1) net = conv2d(net, filters, (3, 3), 1) net = net + x # net=tf.keras.layers.LeakyReLU(0.2)(net) return net def conv2d(inputs, filters, kernel_size, strides): x = tf.keras.layers.Conv2D(filters, kernel_size, strides, 'same')(inputs) x = tf.keras.layers.BatchNormalization()(x) x = tf.keras.layers.LeakyReLU(alpha=0.2)(x) return x d0 = tf.keras.layers.Input(shape=(10, 128, 128, 3)) out= tf.keras.layers.ConvLSTM2D(filters=32, kernel_size=3,padding='same')(d0) out=tf.keras.layers.Conv2D(3,1,1,'same')(out) return keras.Model(inputs=d0, outputs=out)

修改如下：

def build_generator(self):
        def res_net(inputs, filters):
            x = inputs
            net = conv2d(x, filters // 2, (1, 1), 1)
            net = conv2d(net, filters, (3, 3), 1)
            net = net + x
            return net

        def conv2d(inputs, filters, kernel_size, strides):
            x = tf.keras.layers.Conv2D(filters, kernel_size, strides, 'same')(inputs)
            x = tf.keras.layers.BatchNormalization()(x)
            x = tf.keras.layers.LeakyReLU(alpha=0.2)(x)
            return x

        d0 = tf.keras.layers.Input(shape=(10, 128, 128, 3))
        out = tf.keras.layers.ConvRNN2D(filters=32, kernel_size=3, padding='same')(d0)
        out = tf.keras.layers.Conv2D(3, 1, 1, 'same')(out)
        return keras.Model(inputs=d0, outputs=out)

将 `ConvLSTM2D` 替换为 `ConvRNN2D` 即可，它们的参数意义和使用方法都是一样的。