Keras模型加载之通过json文件

最新推荐文章于 2022-05-13 13:45:18 发布

踩着七彩祥云的猴子

最新推荐文章于 2022-05-13 13:45:18 发布

阅读量1.5k

点赞数 1

分类专栏： python进阶知识 Keras

本文链接：https://blog.csdn.net/weixin_36149892/article/details/105661479

版权

python进阶知识同时被 2 个专栏收录

34 篇文章 0 订阅

订阅专栏

Keras

1 篇文章 0 订阅

订阅专栏

导入图

model_from_json

导入权重

load_weights
Note： load_weights没有返回值，千万不要model=model.load_weights(path)

代码示例

import numpy as np
from keras.models import Model, load_model
from keras.models import model_from_json
import json

json_file = '/path/to/model.json'
with open(json_file, 'r') as f:
    json_str = f.read()

## NOTE: 因为模型中有自定义的模块，所以导入json的时候需要添加一个参数custom_objects，一般情况下是不用加的
model = model_from_json(json_str, custom_objects={'BilinearUpSampling2D': BilinearUpSampling2D})
print(type(model))

## NOTE:注意可以只加载weights中有的一些权重，没有的就跳过，可以通过by_name参数实现
model.load_weights(
    '/path/to/AtrousFCN_Resnet50_16sweights.01-0.74.hdf5',
    by_name=True)
print("Load model sucess *****")