在电子应用程序中运行张量流保存的模型

时间:2019-04-13 17:48:38

标签: tensorflow electron

我们有一个自定义的tensorflow保存模型。我们将构建使用带有电子模型的应用程序。

我们如何在电子应用程序中使用此模型?

我们无法将其转换为tensorflow.js模型,我们需要不受支持的功能。

我们的解决方案之一是在macOS上使用仅适用于macos的core-ml引擎,而我们没有针对Windows的解决方案。

感谢任何想法或帮助

更新:

我们将在推理模式下使用模型。

1 个答案:

答案 0 :(得分:0)

理想情况下,您将提供有关如何在Electron中使用Tensorflow的更多详细信息,但是在我的回答中,我假设您只需要运行推理。

在这种情况下,您需要:

  1. 确保已设置python环境(您可以使用virtualenv或类似的东西)
  2. 确保已安装所有需求,例如tensorflow-/second
  3. 通过npm-/first
  4. 安装python shell软件包
  5. 编写一个Python脚本来执行与tensorflow的交互
  6. 使用pyshell执行脚本。这是一个示例:
function App() {
    return (
        <Route exact path={'/fisrt'} component={FirstLayout} /> // FirstLayout already contains Header component
        <Route exact path={'/second'} component={SecondLayout} /> // SecondLayout already contains Header component
        <Route exact path={'/third'} component={ThirdLayout} />
    )
}

最后:另一个选择是用c ++编写与node module API交互的Tensorflow C++