【发布时间】:2021-04-01 00:59:01
【问题描述】:
我制作了一个 cnn,并从第三个 conv 层中提取了特征,因此形状为 (600,12,12,128)
600 = 训练图像 12x12128 = 单张图片
现在我想将这些作为输入传递给另一个 cnn2 但我希望输入形状为 (600,12,12,1)。
如何改变形状?
【问题讨论】:
标签: python keras conv-neural-network feature-extraction
我制作了一个 cnn,并从第三个 conv 层中提取了特征,因此形状为 (600,12,12,128)
600 = 训练图像 12x12128 = 单张图片
现在我想将这些作为输入传递给另一个 cnn2 但我希望输入形状为 (600,12,12,1)。
如何改变形状?
【问题讨论】:
标签: python keras conv-neural-network feature-extraction
尝试使用 Lambda layer 以从 cnn1 分割输出张量:
model=models.Sequential()
model.add(cnn1)
model.add(Lambda(lambda x : x[:,:,i])) # with 0 <= i < 128
model.add(cnn2)
【讨论】: