无法在tensorflow.js中加载python训练的模型

内容来源于 Stack Overflow,并遵循CC BY-SA 3.0许可协议进行翻译与使用

  • 回答 (2)
  • 关注 (0)
  • 查看 (437)

当我使用loadModel()函数tensorflow.js 尝试加载我在Python中训练的模型时,我收到以下错误:

无法加载资源:net :: ERR_NAME_NOT_RESOLVED 未捕获(在承诺中)TypeError:无法获取

以下是predict.js文件

console.log ("hello");

let model;
(async function () {
    model = await tf.loadModel("http://keras_model/model.json");
    $(".progress-bar").hide();
    console.log("it works");
})();

目录结构:

main
  -dataset  (contains images for training the model) 
  -training_scripts (python scripts to train the model)
  -user_interface
     --server.js (server made using node.js(and express))
     --static (this folder contains the trained keras model)
         --index.html (html file to be served)
         --predict.js 
         --keras_model(this folder contains the model.json file)

任何帮助将不胜感激!!

提问于
用户回答回答于

如果要在tfjs中加载本地文件,则需要使用文件类型file:///,为此需要使用tfjs的节点扩展。您可以通过安装并要求加载它来加载它node-fetch

用户回答回答于

您还可以使用fileSystemtfjs-node中公开的处理程序,如下所示:

const tf = require("@tensorflow/tfjs");
const tfn = require("@tensorflow/tfjs-node");
const handler = tfn.io.fileSystem("./path/to/your/model.json");
const model = await tf.loadModel(handler);

扫码关注云+社区

领取腾讯云代金券