将Keras稠密连接到Conv2D输出

2024-10-01 02:29:29 发布

您现在位置:Python中文网/ 问答频道 /正文

假设我有两个要连接的层:(稠密的)称为outg,(Conv2D)称为outm,输出形状为(None,256)和(None,32,32,256)。我的目标是将输出合并到形状中(无、32、32、512)。为了更好地理解,请参见下面的图片。蓝色区域表示Conv2D,橙色区域表示重复的密集层输出。

enter image description here

我的想法是使用来自keras.layersRepeatVector, Reshape and Concatenate,所以我做了:

outg_repeated = RepeatVector(32*32)(outg)
outg_reshaped = Reshape((32, 32, 256))(outg_repeated)
concatenated = Concatenate(axis=-1)([outm, outg_reshaped])

它给出了所需的形状(None,32,32,512),但我不知道是否正确连接了它,这与Conv2D输出垂直,所以我的问题是:它做得对吗,还是我忽略了什么


Tags: none区域目标图片repeated橙色蓝色形状