我想使用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 的情况下获取中间层特征?也就是说,我想提取给定图像的较低级别特征。非常感谢!