小编mat*_*sis的帖子

如何取出aa预训练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')
Run Code Online (Sandbox Code Playgroud)

现在如果我有一批图像

image_batch  =np.ones((5,224,224,3),np.float32)
Run Code Online (Sandbox Code Playgroud)

我可以通过以下方式获得 VGG16 的最后一层

last_layer = vgg16(image_batch)
Run Code Online (Sandbox Code Playgroud)

有谁知道在给定输入图像 image_batch 的情况下获取中间层特征?也就是说,我想提取给定图像的较低级别特征。非常感谢!

python deep-learning keras tensorflow

1
推荐指数
1
解决办法
3238
查看次数

标签 统计

deep-learning ×1

keras ×1

python ×1

tensorflow ×1