lixiangchun/tfjs_resnet18

const tf = require('@tensorflow/tfjs');
const tfnode = require('@tensorflow/tfjs-node');

async function loadModel(){
    const handler = tfnode.io.fileSystem('checkpoint/model.json');
    const model = await tf.loadGraphModel(handler);
    console.log("Model loaded")

    const zeros = tf.randomNormal([8, 3, 224, 224]);
    const out = model.predict(zeros);
    console.log(out);
    out.print();
}

loadModel();