如何修复 "model.predict is not a function" (tensorflow.js)？

Question

我想测试下面的错误代码：

'use strict'

const tf = require('@tensorflow/tfjs');
require('@tensorflow/tfjs-node');
const { createCanvas, createImageData } = require('canvas');

const canvas = createCanvas(800, 600);

async function load_model() {
    let m = await tf.loadLayersModel('path/to/model.json')
    return m;
}

let model = load_model();

model.then(function (res) {
    const example = tf.browser.fromPixels(canvas);
    const prediction = model.predict(example);
    console.log(prediction);
}, function (err) {
    console.log(err);
});

但我有一个）它告诉 TypeError: model.predict is not a function。为什么会这样？我用这个 link 作为教程。

Answer 1

predict 兑现承诺

model.then(function (res) {
    const example = tf.browser.fromPixels(canvas);
    const prediction = res.predict(example);
    console.log(prediction);
}, function (err) {
    console.log(err);
});

如何修复 "model.predict is not a function" (tensorflow.js)？

How to fix "model.predict is not a function" (tensorflow.js)?

javascript

tensorflow.js