微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

“conv_base.output”和“conv_base.input”在这段代码中是什么意思或做什么?

如何解决“conv_base.output”和“conv_base.input”在这段代码中是什么意思或做什么?

conv_base = Xception(include_top=False,input_tensor=None,pooling=None,input_shape=(TARGET_SIZE,TARGET_SIZE,3),classifier_activation='softmax')
                           
model = conv_base.output
model = layers.GlobalAveragePooling2D()(model)
model = layers.Dense(5,activation = "softmax")(model)
model = models.Model(conv_base.input,model)

model.compile(optimizer = Adam(lr = 0.001),loss = "sparse_categorical_crossentropy",metrics = ["acc"])

谁能解释一下这段代码中的 conv_base.outputconv_base.input 是什么意思?它的用途和作用是什么??

解决方法

tf.keras.models.Model 需要 inputsoutputs 作为参数:

参数: 输入:模型的输入:keras.Input 对象或列表 keras.Input 对象。 输出:模型的输出。请参阅下面的函数式 API 示例。 name:字符串,模型名称。

在 Keras 模型中,model.inputmodel.output 返回输入/输出张量,您可以将其作为新模型的输入/输出传递:

from tensorflow.keras.applications.vgg19 import VGG19

base_model = VGG19(weights=None)

print(base_model.input)
Tensor("input_2:0",shape=(None,224,3),dtype=float32)

请注意,调用这些层会返回张量,而不是模型,因此您似乎错误地分配了变量名称。相反,使用以下内容:

model_output = conv_base.output
layer_output = layers.GlobalAveragePooling2D()(model_output)
final_model_output = layers.Dense(5,activation = "softmax")(layer_output)
model = models.Model(conv_base.input,final_model_output)

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。