使用ONNX.js合并两个ONNX模型

时间:2019-10-21 07:38:33

标签: node.js onnx

我正在使用ONNX.js来获得两个经过预训练的ONNX模型。 我想将这两个模型结合起来(使用简单的平均值)来得到一个结合起来的模型,这可以给我带来更好的性能。 使用的两个模型是:https://github.com/microsoft/onnxjs-demo/tree/data/data/test/node/test_addhttps://github.com/microsoft/onnxjs-demo/tree/data/data/test/node/test_add_bcast 我正在使用以下脚本加载两个模型

    //Create an ONNX inference session with WebAssembly backend.
    const session = new onnx.InferenceSession({backendHint:     'wasm'});
    // Load an ONNX model. This model takes two tensors of the same    size and return their sum.
    await session.loadModel('/add.onnx');
    console.log('Model has been loaded');

谢谢

0 个答案:

没有答案