更改KERAS(TF后端)中预训练模型的输入形状

时间:2017-08-28 04:11:30

标签: tensorflow deep-learning keras pre-trained-model

我正在使用带有TensorFlow后端的KERAS的预训练模型进行转移学习。现在,我想改变它的输入形状,但我无法做到。

以下是目前的形状:

model.input_shape : `(None, 16, 112, 112, 3)`

model.layers[0].get_weights(): `(3, 3, 3, 3, 64)`

Output shape of first Layer : `(None, 16, 112, 112, 64)`

我要做的是更改input shape to (None, 112, 112, 112, 1)

我确实尝试更改json文件中的输入,但我收到了错误。

0 个答案:

没有答案