Keras 是源于 Theano 或 者TensorFlow 的一個深度學習框架，它的設計來源于Torch，編程語言使用的是 Python ，是一個擁有強大功能、內容抽象，而且高度模塊化的神經網絡庫。

今天小編給大家分享的就是Keras 模型的保存與加載，希望對大家學習和使用Keras 有所幫助。

一、Keras模型保存和加載的基礎介紹

Keras模型保存和加載一般是保存成hdf5格式。Keras模型主要有兩種，序貫模型即Sequential、以及函數式模型Model，相對來說函數模型Model使用范圍更廣，序貫模型Sequential可看作是函數模型的一種特殊情況。

兩類模型有一些方法是相同的：

model.summary()：打印模型概況

model.get_config():返回包含模型配置信息的Python字典。

model.get_layer()：依據層名或下標獲得層對象

model.get_weights()：返回模型權重張量的列表，類型為numpy array

model.set_weights()：從numpy array里將權重載入給模型，要求數組具有與model.get_weights()相同的形狀。

model.to_json：返回代表模型的JSON字符串，僅包含網絡結構，不包含權值。

model.to_yaml：與model.to_json類似，同樣可以從產生的YAML字符串中重構模型

model.save_weights(filepath)：將模型權重保存到指定路徑，文件類型是HDF5(后綴是.h5)

model.load_weights(filepath, by_name=False)：從HDF5文件中加載權重到當前模型中, 默認情況下模型的結構將保持不變。如果想將權重載入不同的模型(有些層相同)中，則設置by_name=True，只有名字匹配的層才會載入權重

二、Keras模型保存和加載方式

1.保存所有狀態

(1)保存模型和模型圖

# 保存模型 model.save(file_path)
model_name = '{}/{}_{}_{}_v2.h5'.format(params['model_dir'],params['filters'],params['pool_size_1'],params['pool_size_2'])
model.save(model_name)

# 保存模型圖
from keras.utils import plot_model
# 需要安裝pip install pydot
model_plot = '{}/{}_{}_{}_v2.png'.format(params['model_dir'],params['filters'],params['pool_size_1'],params['pool_size_2'])
plot_model(model, to_file=model_plot)

(2)加載模型

from keras.models import load_model

model_path = '../docs/keras/100_2_3_v2.h5'
model = load_model(model_path)

利弊分析：

a.模型保存和加載就只需一行代碼，寫起來簡單快捷

b.既能保存模型的結構和參數，又能保存訓練配置等信息。方便我們從上次訓練中斷的地方再次進行訓練優化。

c.占用空間過大，上傳或者同步費時。

2.只保存模型結構和模型參數

(1)保存模型

import yaml
import json

# 保存模型結構到yaml文件或者json文件
yaml_string = model.to_yaml()
open('../docs/keras/model_architecture.yaml', 'w').write(yaml_string)
# json_string = model.to_json()
# open('../docs/keras/model_architecture.json', 'w').write(json_string)

# 保存模型參數到h5文件
model.save_weights('../docs/keras/model_weights.h5')

(2)加載模型

import yaml
import json
from keras.models import model_from_json
from keras.models import model_from_yaml

# 加載模型結構
model = model_from_yaml(open('../docs/keras/model_architecture.yaml').read())
# model = model_from_json(open('../docs/keras/model_architecture.json').read())

# 加載模型參數
model.load_weights('../docs/keras/model_weights.h5')

利弊分析：

a.能夠節省硬盤空間，便于同步和協作

b.會丟失訓練的一部分配置信息