美文网首页
获取中间层的输出

获取中间层的输出

作者: 梁新彦 | 来源:发表于2017-12-21 22:15 被阅读0次

ref

  1. 要获取中间层的输出,最好的办法是新建一个模型
from keras.models import Model
model = ...  #原始model
layer_name = "my_layer"
intermediate_layer_model = Model(inputs=model.input, outputs=model.get_layer(layer_name).output)
intermediate_output = intermediate_layer_model.predict(data)


####or
    base_model = load_model(model_path)
    base_model.layers.pop()
    pre = Dense(units=num_label,activation='softmax')(base_model.layers[-1].output)
    model = Model(base_model.input,pre)
  1. 或者使用keras function来实现返回一个特定的输出
from keras import backend as K
get_3rd_layer_output = K.function([model.layers[0].input, model.layers[3].output])
layer_output = get_3rd_layer_output([x])[0]
  1. 固定特定层权重freeze weights
base_model = InceptionV3(weights='imagenet', include_top=False) 
for layer in base_model.layers:  
    layer.trainable = False

或者比如指定前3层不训练

for layer in base_model.layers[:3]:  
   layer.trainable = False
  1. 使用pop方法来删除最后一层
model = Sequential()
model.add(Dense(32, activation="relu", input_dim=784))
model.add(Dense(32, activation="relu"))
print(len(model.layers)) #输出2
model.pop()
print(len(model.layers))  #输出1

相关文章

  • 获取中间层的输出

    ref 要获取中间层的输出,最好的办法是新建一个模型 或者使用keras function来实现返回一个特定的输出...

  • Keras 获取中间层输出

    一种简单的方法是创建一个新的Model,使得它的输出是你想要的那个输出 也可以建立一个Keras的函数来达到这一目...

  • PyTorch 获取模型中间层方法

    获取模型中间层 通过index获取 通过特征名获取

  • MXNet-获取中间层输出及相应的梯度

    最近为了得到CNN中每一层激活值对应的梯度(自动计算的梯度只有每一层对应的权值和偏置,MX又没有pytorch那样...

  • pytorch 中间层输出

    想获取网络的中间输出,但是尝试后,发现 hook不好用 Sequential有时用不了 所以最终决定还是直接使用l...

  • keras输出中间层结果

    转载自keras输出中间层结果的2种方法 使用函数模型API,新建一个model,将输入和输出定义为原来的mode...

  • keras输出中间层结果

    1.使用函数模型API,新建一个model,将输入和输出定义为原来的model的输入和想要的那一层的输出,然后重新...

  • 适配器模式

    1. 模式理解 提供一个adpater中间层,这个中间层的作用就是把不统一的输入转换成统一的输出 2. 解决一个什...

  • Chapter3_神经网络

    神经网络 输入层(第0层)->中间层(隐藏层)->输出层(最后一层) 隐藏层的激活函数 将输入信号的总和转换为输出...

  • Pytorch 随记 (看了书籍随手记录,无逻辑无重点所在)

    在前向传播函数(forward)函数中,有意识的将输出变量都命名为x,是为了能让Python回收一些中间层的输出,...

网友评论

      本文标题:获取中间层的输出

      本文链接:https://www.haomeiwen.com/subject/lbszwxtx.html