javascript - 如何修复 "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 是基于那时返回的 promise

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