毕设(二)：载入模型函数

最新推荐文章于 2022-05-20 10:24:34 发布

DC妙妙屋

最新推荐文章于 2022-05-20 10:24:34 发布

阅读量424

点赞数

本文链接：https://blog.csdn.net/qq_19875729/article/details/115322689

版权

定义了一个名为`load_model`的函数，用于加载保存在`model_path`路径下的LeNet5模型。模型首先被初始化，然后配置损失函数为`categorical_crossentropy`，优化器为`adam`，并添加了精度作为评估指标。权重从指定路径加载，并打印模型摘要。最后返回加载完成的模型。

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

写一个载入模型的函数

#creat model
def load_model(model_path = './model/modelckpt')
    model = LeNet5()
    model.build(input_shape = (None,64,64,1))
    model.compile(loss = 'categorical_crossentropy', optimizer = 'adam' , metrics = ['accuracy'])
    model.load_weights(model_path)
    model.summary()
    return model