mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-14 20:48:00 +00:00
cd8775f518
### Description The Node JS Samples included in the repository have outdated package references that are broken, which are fixed in this PR. ### Motivation and Context The samples included in this repository should just work, but sadly do not. The reason is that they are using very outdated references for the npm modules. This fix updates the dependencies to the current onnxruntime-node, which fixes the samples. Also adds a small update to the .gitignore to exclude the node_modules directories in the samples directory, which keeps the local repo changelist cleaner.
38 lines
1.5 KiB
JavaScript
38 lines
1.5 KiB
JavaScript
'use strict';
|
|
const fs = require('fs');
|
|
const util = require('util');
|
|
const InferenceSession = require('onnxruntime-node').InferenceSession;
|
|
|
|
// use an async context to call onnxruntime functions.
|
|
async function main() {
|
|
try {
|
|
// session options: please refer to the other example for details usage for session options
|
|
const options = { intraOpNumThreads: 1 };
|
|
|
|
//
|
|
// create inference session from a ONNX model file path
|
|
//
|
|
const session01 = await InferenceSession.create('./model.onnx');
|
|
const session01_B = await InferenceSession.create('./model.onnx', options); // specify options
|
|
|
|
//
|
|
// create inference session from an Node.js Buffer (Uint8Array)
|
|
//
|
|
const buffer02 = await util.promisify(fs.readFile)('./model.onnx'); // buffer is Uint8Array
|
|
const session02 = await InferenceSession.create(buffer02);
|
|
const session02_B = await InferenceSession.create(buffer02, options); // specify options
|
|
|
|
//
|
|
// create inference session from an ArrayBuffer
|
|
//
|
|
const arrayBuffer03 = buffer02.buffer;
|
|
const offset03 = buffer02.byteOffset;
|
|
const length03 = buffer02.byteLength;
|
|
const session03 = await InferenceSession.create(arrayBuffer03, offset03, length03);
|
|
const session03_B = await InferenceSession.create(arrayBuffer03, offset03, length03); // specify options
|
|
} catch (e) {
|
|
console.error(`failed to inference ONNX model: ${e}.`);
|
|
}
|
|
}
|
|
|
|
main();
|