如何使用keras将高级功能与CNN中的低级功能连接起来?

时间:2017-07-19 11:43:40

标签: python machine-learning neural-network keras deep-learning

我正在尝试将一个图层的输出传递并连接到CNN中的另一个图层,如下图所示。重塑后,30x30x512层与15x15x1024连接。

enter image description here

请注意,两种不同的型号并合并它们并不是我想要的解决方案。我想知道如何使用相同的模型来合并提到的图层。

提前感谢您的建议。

1 个答案:

答案 0 :(得分:3)

您需要存储要合并的两个图层的输出张量。拥有这些张量后,至少有两种方法可以合并它们:

  1. 使用Reshape图层,您可以在其中指定输出形状并完成作业。

  2. 使用tf.space_to_depth函数并将其打包到Lambda图层(因为它是TensorFlow操作)。这可能仅在您使用TensorFlow后端时使用。

  3. 重新塑造输出后,您可以使用concatenate图层并按渠道尺寸合并输出。