在本地将 keras model 加载到 tensorflow.js 中

Question

我想加载已转换为tensorflow.js兼容格式的keras model 进行推理。 我的代码看起来像这样

import * as tf from '@tensorflow/tfjs';
import "regenerator-runtime/runtime.js";
import 'bootstrap/dist/css/bootstrap.css';

const model_path = '/home/user/Desktop/Github/tfjs_model/tfjs/tfjs_model/model.json'; 

async function loadModel(path){
  console.log("Model loading in progress from ".concat(path));
  const model =  await tf.loadLayersModel(path);
  console.log("Model Loaded Successfully");
  return model;
};

const model = loadModel(model_path);

const input = tf.tensor1d([1], [13]);
console.log(model.predict(input));

我收到此错误Error: Failed to parse model JSON of response from /home/user/Desktop/Github/tfjs_model/tfjs/tfjs_model/model.json'. Please make sure the server is serving valid JSON for this request. Error: Failed to parse model JSON of response from /home/user/Desktop/Github/tfjs_model/tfjs/tfjs_model/model.json'. Please make sure the server is serving valid JSON for this request. 我查看了很多地方（Github 问题，堆栈溢出），但似乎找不到解决方法。 如何解决？ `

Answer 1

我也有这个问题。

使用const model = await tf.loadLayersModel('directory/model.json'); 产生了与您相同的错误。 我尝试更改导入以查找文件model2.json ，该文件尚不存在但给出了相同的错误（我注意到 TensorFlow.js 存在一个突出问题以给出更有意义的错误）。

查看我的 Parcel 设置，我意识到model.json和.bin文件被复制到dist文件夹的根目录中。 所以我从导入中删除了目录，现在它可以使用：

const model = await tf.loadLayersModel('model.json');

请注意，我的文件仍在<content_root>/directory/中。