在单程中将两个图像作为输入,将两个图像作为输出?

Ham*_*mza 5 artificial-intelligence machine-learning autoencoder deep-learning unity3d-unet

我正在构建一个 UNET 模型。我陷入了一件事,如果我们必须将两个图像作为输入传递给 UNET,我们可以通过连接它们来完成它,就像下面的代码

inputs = []
for _ in range(num_inputs):
    inputs.append(Input((self.input_height, self.input_width, self.input_features)))
x = concatenate(inputs)
x = Conv2D(32, (3, 3), activation='relu', padding='same')(x)
Run Code Online (Sandbox Code Playgroud)

但是我们如何才能在一次通过中获得两个图像作为输出?