[英]Keras - get weight of trained layer
I'm trying to get the values of a layer in a trained network. 我正在尝试在训练有素的网络中获取图层的值。 I can get the layer as a TensorFlow Tensor, but I'm unable to access its values in an array shape:
我可以将图层作为TensorFlow Tensor获取,但我无法以数组形状访问其值:
from keras.models import load_model
model = load_model('./model.h5')
layer_dict = dict([(layer.name, layer) for layer in model.layers])
layer_name = 'block5_sepconv1_act'
filter_index = 0
layer_output = layer_dict['model_1'][layer_name].output
# <tf.Tensor 'block5_sepconv1_act/Relu:0' shape=(?, 16, 16, 728) dtype=float32>
layer_filter = layer_output[:, :, :, filter_index]
# <tf.Tensor 'strided_slice_11:0' shape=(?, 16, 16) dtype=float32>
# how do I get the 16x16 values ??
If you use the tensorflow
backend, you can evaluate the value of a tensor using the current session sess
and feeding the correct input 如果使用
tensorflow
后端,则可以使用当前会话sess
评估张量值并输入正确的输入
import keras.backend as K
input_value = np.zeros(size=(batch_size, input_dim))
sess = K.get_session()
output = sess.run(layer_output, feed_dict={model.input: input_value})
If you just want to retrieve the weights, you can evaluate the weights of a layers using: 如果您只想检索权重,可以使用以下方法评估图层的权重:
weights = [w.eval(K.get_session) for w in layer_dict['model_1'][layer_name].weights]
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.