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

Question

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

tf.loadModel(Model_URL, Weights_URL)

但是给这个文件路径会抛出这个错误:

URL scheme "file" is not supported.

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

Answer 1

试试下面的代码行

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