如何制作具有不同输入形状的批次的神经网络

Question

我想制作一个可以将灰度图像作为输入并输出彩色图像的CNN或FCN。对我来说非常重要的是图像的大小可以变化。我听说只有在制作FCN并以一个尺寸的图像进行批处理而另一个以另一尺寸的图像进行批处理时才能这样做。但是我不知道如何在Tensorflow Keras（Python版本）中提出这个概念，我想知道您是否可以提供一些示例代码或伪代码？我很感激。谢谢！

Answer 1

我知道您想将它们全部保留为原始大小，但这是不可能的。不过，请不要担心，因为调整大小可以在将图像馈入模型时（在内存中）进行；除了被读取之外，图像永远不会被触摸。

我经常引用的

Here's a great example！