在本地加载一个 Tensorflow.js 模型

Question

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

<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>

Directory of my GitHub Repository

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

Answer 1

Tensorflow.js 无法直接访问本地文件（除非您在 Node.js 中进行）。看看these answers的一些内容。

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

1. 从 HTTP 服务器加载模型
2. 从HTML file input elements 的用户选择的文件中加载模型。

Answer 2

将此行 const model = await tf.loadLayersModel('AI-Model/model.json'); 替换为 const model = await tf.loadGraphModel('http:AI-Model/model.json');