無法通過tensorflow.js model（）加載本地模型

Question

我有一個模型，我將其轉換為tensorflowjs的json。 現在，我試圖從本地加載此模型，但是無法加載它，並且出現錯誤Failed to load locally-saved model. TypeError: Cannot read property 'sourceLayer' of undefined 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);