python - 如何删除 Inception 模型的最后一层

Question

嗨，我关注了 Github 上关于删除预训练模型的最后一层的论坛。但是它对我不起作用；也许我做错了什么

我正在关注这个，这是我的代码。我以为我要做model.layers[-2].output的就是，但它告诉我AttributeError: 'Tensor' object has no attribute 'summary'

import tensorflow as tf
import tensorflow.keras as keras
import tensorflow.keras.backend as K
import numpy as np
from tensorflow.keras.layers import Dense, Input, Layer
from tensorflow.keras.models import Model
from tensorflow.keras.applications.inception_v3 import InceptionV3


model = InceptionV3()
print(model.summary())

modele = model.layers[-2].output
print(modele.summary())

标签： pythontensorflowkeras

Answer 1

modele 变量只是一层。你所要做的：

model = InceptionV3()
print(model.summary())

output = model.layers[-2].output
modele = Model(inputs = model.input, outputs = output)
print(modele.summary())