无法通过tensorflowjs model()加载本地模型

Question

我有一个模型，我将其转换为tensorflowjs的json。现在，我尝试从本地加载此模型，但无法加载，并出现错误Failed to load locally-saved model. TypeError: Cannot read property 'sourceLayer' of undefined

我使用nodev8.16.0，tensorflow/tfjs:"^1.2.6“和@tensorflow/tfjs-node："^1.2.5”。我尝试使用各种方法加载模型tf.model() tf.LayersModel()

我还尝试使用tfn.io.fileSystem加载模型

global.fetch = require('node-fetch');
const tf = require('@tensorflow/tfjs');
const tfn = require("@tensorflow/tfjs-node");

async function init() {
    const LOCAL_MODEL_PATH = 'file://some_model.json';
    // const handler = tfn.io.fileSystem(LOCAL_MODEL_PATH);
    let model;
    try {
        model = await tf.model(LOCAL_MODEL_PATH);
        model.summary();
        console.log('Loaded locally-saved model!');
        // runAndVisualizeInference(model);
    } catch (err) {
        console.log('Failed to load locally-saved model. ' + err);
    }
}
init();

Answer 1

tf.model的参数是一个符号张量对象。由于未使用正确的参数，因此会引发错误。值得注意的是，tf.model用于构建新模型。这里要做的是加载保存在model.json中的现有模型。

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