Keras学习001:keras读取模型某一层的输出

keras提供很多训练好的模型vgg16,vgg19,InceptionV3等如何获取这些模型的某一层的输出 这些模型每个模型都有超过10个层,如果想检查中间的某层输出如何操作

1. 加载模型

以vgg16为例

from keras.applications.vgg16 import VGG16

model = VGG16(weights='imagenet', include_top=False)

print(model.layers)

如果第一次使用vgg16则模型需要一段时间下载,下载完成下次就会直接加载 下载完保存在(~/.keras/models/)目录加载完可以看到输出一个list每一项为一layer

2. 定义输出中间层layer模型

下面代码定义个输出第3层的模型

from keras.models import Model
layer_model = Model(inputs=model.input,outputs=model.layers[3].output)

定义好的layer_model就是输出第3层

3. 完整的测试代码

from keras.applications.vgg16 import VGG16
from keras.preprocessing import image
from keras.applications.vgg16 import preprocess_input
from keras.models import Model
import numpy as np
import pylab

base_model = VGG16(weights='imagenet')
layer_model = Model(inputs=model.input, outputs=model.layers[3].output)

img_path = 'elephant.jpg'
img = image.load_img(img_path, target_size=(224, 224))
x = image.img_to_array(img)
x = np.expand_dims(x, axis=0)
x = preprocess_input(x)

features = model.predict(x)

print(features.shape)
pylab.imshow(features[0][:,:,0])
pylab.show()
  • 4
    点赞
  • 20
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值