-
actix-tensorflow-example: 示例展示了如何通过Actix Web服务利用TensorFlow的锈绑定来部署训练好的机器学习模型
资源介绍
使用Actix-Web服务TensorFlow
该存储库提供了一个示例,该示例使用python中的TensorFlow2.0 Keras训练机器学习模型,然后导出该模型,然后使用Rust中的Actix-Web通过RESTful API进行推理。
该示例背后的动机是在一个简单而实际的用例中尝试。
有关使用的工具的更多信息,请查看以下存储库:
概述
该存储库包含3个部分:
./training –包含训练神经网络以识别数字的脚本。
./server –包含RESTful API Web服务器rust代码。
./client –包含一个示例脚本,该脚本演示了如何向服务器发出请求。
训练脚本将以TensorFlow的protobuf格式输出保存的神经网络模型。 然后,服务器在启动时将其加载到内存中。 服务器在/mnist端点接受带有单个密钥“图像”的JSON负载,该密钥是base64