如何修改keras应用程序中的内置模型

时间:2019-01-15 23:00:14

标签: keras

我需要从keras model的多个层获取多个输出,而不是从最后一层获取输出。

我知道如何根据我的需要调整代码,但是不知道如何在keras application中使用它。我的意思是我该如何导入它。我是否需要再次陷入setup.py的{​​{1}}。我这样做了,但是什么也没发生。我的更改未应用。有没有其他方法可以从模型中的不同层获取输出?

1 个答案:

答案 0 :(得分:1)

实际上,解决方案是如此简单,您只需要调用指定图层的输出即可。

new_model= tf.keras.Model(base_model.input, base_model.get_layer(layer_name).output)