python - 如何取出aa pretrained keras模型的中间层
问题描述
我想使用VGG
模型(tensorflow 或 keras 预训练模型)作为特征提取器;我加载VGG16 model
:
IMG_SHAPE = (224, 224, 3)
vgg16 = tf.keras.applications.VGG16(input_shape = IMG_SHAPE,
include_top=False,
weights='imagenet')
现在如果我有一批图像
image_batch =np.ones((5,224,224,3),np.float32)
我可以通过
last_layer = vgg16(image_batch)
有谁知道在给定输入图像 image_batch 的情况下获得中间层特征?那就是我想提取给定图像的较低级别的特征。非常感谢!
解决方案
您可以执行以下操作:
IMG_SHAPE = (224, 224, 3)
model = tf.keras.applications.VGG16(input_shape = IMG_SHAPE,
include_top=False,
weights=None)
pretrain_model_path = "weights/vgg16_weights_tf_dim_ordering_tf_kernels_notop.h5"
model.load_weights(pretrain_model_path)
# print(model.summary())
image_batch = np.ones((5,224,224,3),np.float32)
last_layer = tf.keras.models.Model(inputs=model.input, outputs=model.get_layer('block5_pool').output)
res = last_layer.predict(image_batch)
但是,你怎么知道要传入model.get_layer()
什么?
答案 - 通过model.summary()
如果您打印 的输出model.summary()
,您将获得不同的图层名称,您可以将它们传入model.get_layer()
并获取该图层的输出。
Layer (type) Output Shape Param #
=================================================================
input_17 (InputLayer) (None, 224, 224, 3) 0
_________________________________________________________________
block1_conv1 (Conv2D) (None, 224, 224, 64) 1792
_________________________________________________________________
block1_conv2 (Conv2D) (None, 224, 224, 64) 36928
_________________________________________________________________
block1_pool (MaxPooling2D) (None, 112, 112, 64) 0
_________________________________________________________________
block2_conv1 (Conv2D) (None, 112, 112, 128) 73856
_________________________________________________________________
block2_conv2 (Conv2D) (None, 112, 112, 128) 147584
_________________________________________________________________
block2_pool (MaxPooling2D) (None, 56, 56, 128) 0
_________________________________________________________________
block3_conv1 (Conv2D) (None, 56, 56, 256) 295168
_________________________________________________________________
block3_conv2 (Conv2D) (None, 56, 56, 256) 590080
_________________________________________________________________
block3_conv3 (Conv2D) (None, 56, 56, 256) 590080
_________________________________________________________________
block3_pool (MaxPooling2D) (None, 28, 28, 256) 0
_________________________________________________________________
block4_conv1 (Conv2D) (None, 28, 28, 512) 1180160
_________________________________________________________________
block4_conv2 (Conv2D) (None, 28, 28, 512) 2359808
_________________________________________________________________
block4_conv3 (Conv2D) (None, 28, 28, 512) 2359808
_________________________________________________________________
block4_pool (MaxPooling2D) (None, 14, 14, 512) 0
_________________________________________________________________
block5_conv1 (Conv2D) (None, 14, 14, 512) 2359808
_________________________________________________________________
block5_conv2 (Conv2D) (None, 14, 14, 512) 2359808
_________________________________________________________________
block5_conv3 (Conv2D) (None, 14, 14, 512) 2359808
_________________________________________________________________
block5_pool (MaxPooling2D) (None, 7, 7, 512) 0
=================================================================
Total params: 14,714,688
Trainable params: 14,714,688
Non-trainable params: 0
推荐阅读
- javascript - 当我使用`.call`方法设置上下文时,为什么嵌套函数中的`this`是窗口对象?
- r - R:使用棘手的分隔符将一列分成几行
- java - zxing-android-embedded 每次首次应用启动时的错误消息
- java - 如何更改我的代码以提高效率?
- c# - 如何在 C# 中引用父类使用的接口?
- ruby-on-rails - 如何使用 Rails 6.1 多数据库 ActiveRecord::ConnectionHandling 连接到主数据库和分片?
- javascript - 如何将 Firebase 对象置于响应状态?
- java - 在 jqwik 属性测试框架中寻找更好的方法来生成图的边列表
- call - 呼叫中心算法 MIA(最空闲代理)和 LOA(最少占用代理)
- python - 将 List 值的 Pandas 列转换为带有计数的标题