如何将从 ccn1 提取的特征传递给另一个 cnn2？

Question

我制作了一个 cnn，然后从第三个 conv 层中提取了特征，使形状为 (600,12,12,128)

600 = 训练图像 12x12128 = 单张图片

现在我想将这些作为输入传递给另一个 cnn2 但我希望输入形状为 (600,12,12,1)。

如何改变形状？

Answer 1

尝试使用 Lambda layer 对来自 cnn1 的输出张量进行切片：

model=models.Sequential()

model.add(cnn1)

model.add(Lambda(lambda x : x[:,:,i])) # with 0 <= i < 128

model.add(cnn2)