如何分发训练好的模型?

4

我希望能够分发使用Keras训练的深度学习模型。

合作伙伴想要在不进行复杂环境配置(例如安装Python、Keras等)的情况下使用它。

我该怎么做?是否需要制作一个.exe文件?


请了解cx_freeze。 - DJK
你能更详细地解释一下你的合作伙伴将如何使用它吗?他们将输入什么类型(以及什么数量)的数据到模型中?他们期望如何接收输出? - Imran
基本解决方案是使用model.save(),如当前答案所述。但这仍需要一些Python依赖。如果您想使其更加自包含,您需要决定您想要什么(服务、API、批处理程序)- 这将定义具体的解决方案。 - Ophir Yoktan
@Imran 我的合作伙伴将把它作为组件添加到他们的软件中,用于图像识别。他们会提供给我训练数据,我对模型进行训练并将训练好的模型交给他们。训练好的模型接收一个或多个图像,并输出分类结果。 - user8483278
1个回答

1

也许您想查看这个tutoiral

最简单的代码:

from keras.models import load_model

model.save('my_model.h5')  # creates a HDF5 file 'my_model.h5'

# returns a compiled model
# identical to the previous one
model = load_model('my_model.h5')

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接