【keras】加载min_xception前面几个层，把后面的层给改成自己需要层。或者修改后面几层的参数

最新推荐文章于 2024-06-23 23:42:44 发布

快乐成长吧

最新推荐文章于 2024-06-23 23:42:44 发布

阅读量1.3k

点赞数

分类专栏： Keras

本文链接：https://blog.csdn.net/qq_37791134/article/details/104577423

版权

本文介绍了如何在Keras中加载预训练的InceptionV3模型，并根据需求修改其后部结构，如替换或调整层的参数。参考了两个教程，包括从InceptionV3进行微调和去掉全连接层的方法。

摘要由CSDN通过智能技术生成

参考：keras预训练模型应用(4)：fine-turn InceptionV3 https://www.jianshu.com/p/23295376c44d

keras调用自己训练的模型，并去掉全连接层 https://blog.csdn.net/qq_29462849/article/details/83010854

from keras.models import load_model
from keras.utils import plot_model
from keras.models import Model
from keras.layers import Activation, Conv2D
from keras.layers import GlobalAveragePooling2D

pre_model = load_model(load_model_path)  # mini_XCEPTION
pre_model.summary()
plot_model(pre_model, to_file='mini_XCEPTION_7.png')  # 画出模型结构图，并保存成图片

base_model = Model(inputs=pre_model.input, outputs=pre_model.get_layer('add_4').output)
base_model.summary()
# 【1】增加一个卷积层
x = base_model.output
x = Conv2D(num_classes, (3, 3),
           # kernel_regularizer=regularization,
           padding