tensorflow2保存和调用训练好的模型

保存和调用整个模型

调用整个模型包括调用模型结构和模型参数，只需保证模型输入符合要求就行。

保存模型

model.save('model.h5')
# model为要保存的模型对象
# model.h5为模型保存文件名和路径

调用模型

model = tf.keras.models.load_model('model.h5', compile=False)
model.compile(optimizer=tf.keras.optimizers.Adam(lr=0.01),
                loss=tf.keras.losses.Huber()
                )
# model调用模型后的模型名
# model.h5为模型保存文件名和路径

保存和调用模型参数

这种形式的调用需要当前模型的所有结构和调用的模型结构一致。

保存模型参数

model.save_weights('weights.ckpt')
# model为要保存参数的模型对象
# weights.ckpt为模型参数保存的文件名和路径

调用模型参数

model.load_weights('weights.ckpt')
# model为本地的模型对象
# weights.ckpt为被调用的模型参数保存的文件名和路径