同样是保存模型，model.save()和model. save_weights ()有何区别

淡淡的烟草味﹌ 2022-12-19 13:28 187阅读 0赞

**model.save()**保存了模型的图结构和模型的参数，保存模型的后缀是.hdf5。

**model. save\_weights** ()只保存了模型的参数，并没有保存模型的图结构,保存模型的后缀使用.h5。

所以使用save\_weights保存的模型比使用save() 保存的模型的大小要小。同时加载模型时的方法也不同。model.save()保存了模型的图结构，直接使用load\_model()方法就可加载模型然后做测试，例：

from  tensorflow.keras.models import load\_model

model=load_model("my_model_.hdf5")

加载save\_weights保存的模型就稍微复杂了一些，还需要再次描述模型结构信息才能加载模型。例：

def bn\_prelu(x):

x = BatchNormalization(epsilon=1e-5)(x)

x = PReLU()(x)

return x

def build\_model(out\_dims, input\_shape=(norm\_size, norm\_size, 3)):

inputs\_dim = Input(input\_shape)

x = Conv2D(32, (3, 3), strides=(2, 2), padding='same')(inputs\_dim)

x = bn\_prelu(x)

x = Conv2D(32, (3, 3), strides=(1, 1), padding='same')(x)

x = bn\_prelu(x)

x = MaxPooling2D(pool\_size=(2, 2))(x)

x = Conv2D(64, (3, 3), strides=(1, 1), padding='same')(x)

x = bn\_prelu(x)

x = Conv2D(64, (3, 3), strides=(1, 1), padding='same')(x)

x = bn\_prelu(x)

x = MaxPooling2D(pool\_size=(2, 2))(x)

x = Conv2D(128, (3, 3), strides=(1, 1), padding='same')(x)

x = bn\_prelu(x)

x = Conv2D(128, (3, 3), strides=(1, 1), padding='same')(x)

x = bn\_prelu(x)

x = MaxPooling2D(pool\_size=(2, 2))(x)

x = Conv2D(256, (3, 3), strides=(1, 1), padding='same')(x)

x = bn\_prelu(x)

x = Conv2D(256, (3, 3), strides=(1, 1), padding='same')(x)

x = bn\_prelu(x)

x = GlobalAveragePooling2D()(x)

dp\_1 = Dropout(0.5)(x)

fc2 = Dense(out\_dims)(dp\_1)

fc2 = Activation('softmax')(fc2) \#此处注意，为sigmoid函数

model = Model(inputs=inputs\_dim, outputs=fc2)

return model

model=build\_model(labelnum)

model. load\_weights(“my\_model\_.h5”);

发表评论取消回复

表情：

评论列表（有 0 条评论，187人围观）

还没有评论，来说两句吧...

相关阅读

相关 Python和Java有何区别

转自：[微点阅读][Link 1] [https://www.weidianyuedu.com][Link 1] 初学编程的小伙伴在问：“Python和Java到底有什么区别

待我称王封你为后i/ 2024年03月30日 16:24/ 0 赞/ 65 阅读

相关 SOA 和微服务有何区别？

![format_png][] 玩过 Dubbo 的小伙伴应该都有听说过一个概念叫做 SOA，每当我们说起微服务的时候，很多人就会去纠结这和 SOA 有啥关系呀？感觉换汤不换

亦凉/ 2023年10月03日 23:24/ 0 赞/ 17 阅读

相关 SEO与SEM是什么？有何区别？

搜索引擎网络营销推广中被问到很多的就是： SEO和SEM的区别是什么？哪一个的转化率更高？怎样使用成本消费更少？如何运用为推广营销带来更积极影响

旧城等待，/ 2023年09月23日 20:28/ 0 赞/ 75 阅读

相关同样是保存模型，model.save()和model. save_weights ()有何区别

model.save()保存了模型的图结构和模型的参数，保存模型的后缀是.hdf5。 model. save\_weights ()只保存了模型的参数，并没有保存模型的图结构

淡淡的烟草味﹌/ 2022年12月19日 13:28/ 0 赞/ 188 阅读

相关 TableAdapter和DataAdapter有何区别

在使用DataAdapter时,需要将查出的数据起一个表名放到DATASET中,一个DATASET可以存放多个表,而TableAdapter的结果就是一个表,不能在继续添加表.

水深无声/ 2022年08月07日 10:48/ 0 赞/ 199 阅读

相关 Java - 什么是领域模型(domain model)？贫血模型(anaemic domain model)和充血模型(rich domain model)有什么区别？

分享一个大牛的人工智能教程。零基础！通俗易懂！风趣幽默！希望你也加入到人工智能的队伍中来！请点击[http://www.captainbed.net][http_www.cap

男娘i/ 2022年06月06日 12:10/ 0 赞/ 363 阅读

相关 Java | Exception 和 Error 有何区别？

> 微信公众号：一个优秀的废人 > 如有问题或建议，请后台留言，我会尽力解决你的问题。前言这周身体不适请了两天病假，只上了三天班。本来不打算发文，但写文章这件事

水深无声/ 2022年03月27日 07:56/ 0 赞/ 263 阅读

相关同样都是程序员，互联网与外包公司有何区别？结果很现实！

随着互联网的发展，程序员工作的形式越来越多样。在面临就业时，一部分程序员选择互联网公司，一部分选择外包公司。从本质上说，无论是互联网公司还是外包公司，都是以营利为目的，但是

我就是我/ 2022年01月12日 04:47/ 0 赞/ 122 阅读

相关 OLTP和OLAP有何区别

OLTP和OLAP主要区别有： 1、基本含义不同：OLTP是传统的关系型数据库的主要应用，主要是基本的、日常的事务处理，记录即时的增、删、改、查，比如在银行存取一笔款，就是一

骑猪看日落/ 2022年01月07日 21:57/ 0 赞/ 361 阅读

相关 Unicode 和 UTF-8 有何区别？

作者：于洋链接：https://www.zhihu.com/question/23374078/answer/69732605 来源：知乎著作权归作者所有。商业

柔情只为你懂/ 2021年06月24日 15:58/ 0 赞/ 451 阅读