Объедините две модели ONNX, используя ONNX.js - PullRequest
Новая
       15

Объедините две модели ONNX, используя ONNX.js

0 голосов
/ 21 октября 2019

Я использую ONNX.js, чтобы получить две предварительно обученные модели ONNX. Я хочу объединить эти две модели (используя простое среднее), чтобы получить комбинированную модель, которая дает мне лучшую производительность. Используются две модели: https://github.com/microsoft/onnxjs-demo/tree/data/data/test/node/test_add и https://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');

Спасибо

...