在本地加载 Tensorflow.js 模型

Question

目前，我正在尝试加载一个 json 模型，该模型已从 tensorflow 转换为我的 tensorflow 函数。

<script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@2.0.0/dist/tf.min.js">
</script>
<script>
    //!pip install tensorflowjs;
    // * as tf from '@tensorlowjs/tfjs';
    async function tensorFlow(){
    const model = await tf.loadLayersModel('AI-Model/model.json');

}

tensorFlow()

</script>

我的 GitHub 存储库目录

但是，当我尝试添加 json 文件时，出现 loadLayersModel() 需要 http 请求的错误。 有没有办法从存储库加载我的模型？ 如果不是，我该怎么做？

Answer 1

Tensorflow.js 不能直接访问本地文件（除非你在 Node.js 中这样做）。 看看其中一些答案。

查看Tensorflow.js API ，您的选择似乎是：

1. 从 HTTP 服务器加载模型
2. 从用户从HTML 文件输入元素中选择的文件加载模型。

Answer 2

替换这一行const model = await tf.loadLayersModel('AI-Model/model.json'); 通过const model = await tf.loadGraphModel('http:AI-Model/model.json');