Tensorflow CNN模型 - RESTful API

时间:2017-12-05 16:14:26

标签: python rest machine-learning tensorflow

我在python中开发了一种卷积神经网络算法,该算法使用特定标签对图像(.jpg)进行分类,

1)定义自定义CNN模型;

2)设置一个估算器,在本地保存save_summary和save_checkpoint步骤;

3)使用estimator.train函数训练估计量。

现在,如果我使用新图像运行estimator.predict函数,它将返回预测标签。

如何将此经过培训的估算器部署为RESTful API,以便我可以从WEB页面或应用程序调用它?

1 个答案:

答案 0 :(得分:0)

了解您是否使用了一些已知的框架(Keras,Tensorflow,MXNet ......)会有所帮助,因为大多数都有一些推荐的方法来通过API提供模型。

如果您从头开始构建解决方案,那么您可以"只是"使用任何Web框架来提供模型预测。为了帮助您入门,您可能需要查看Flask