我需要从keras model
的多个层获取多个输出,而不是从最后一层获取输出。
我知道如何根据我的需要调整代码,但是不知道如何在keras application
中使用它。我的意思是我该如何导入它。我是否需要再次陷入setup.py
的{{1}}。我这样做了,但是什么也没发生。我的更改未应用。有没有其他方法可以从模型中的不同层获取输出?
答案 0 :(得分:1)
实际上,解决方案是如此简单,您只需要调用指定图层的输出即可。
new_model= tf.keras.Model(base_model.input, base_model.get_layer(layer_name).output)