使用具有tensorflow js的图形文件(.pb文件)进行预测

时间:2018-06-19 07:23:46

标签: javascript tensorflow tensorflow.js

我刚开始使用TensorFlow.JS。 我有一个样本" .pb"我的目录中的文件,这是一个训练有素的模型。我尝试加载此模型,以便在我尝试在浏览器中构建的示例计算机学习应用程序中使用它进行预测。由于我对本主题的理解非常少,因此在文档中找不到任何有用的信息。我尝试过使用

tf.loadModel(Model_URL, Weights_URL)

但是给这个文件路径引发了这个错误:

  

网址方案"文件"不受支持。

任何人都可以帮我在浏览器中使用Tfjs加载这个.pb文件吗?

1 个答案:

答案 0 :(得分:1)

尝试以下代码行

const model = await tf.loadFrozenModel(Model_URL, Weights_URL);