How to use the onnxjs.InferenceSession function in onnxjs
To help you get started, we’ve selected a few onnxjs examples, based on popular ways it is used in public projects.
Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
microsoft / onnxjs / benchmark / src / index.js View on Github 
async init(backend, modelPath, imageSize) {
this.imageSize = imageSize;
const hint = {backendHint: backend };
this.model = new onnx.InferenceSession(hint);
await this.model.loadModel(modelPath);
}
async runModel(data) {this.session = this.gpuSession;
return;
}
this.modelLoading = true;
this.modelInitializing = true;
this.gpuSession = new InferenceSession({backendHint: this.sessionBackend});
this.session = this.gpuSession;
}
if (this.sessionBackend === 'wasm') {
if (this.cpuSession) {
this.session = this.cpuSession;
return;
}
this.modelLoading = true;
this.modelInitializing = true;
this.cpuSession = new InferenceSession({backendHint: this.sessionBackend});
this.session = this.cpuSession;
}
try {
await this.session!.loadModel(this.modelFile);
} catch (e){
this.modelLoading = false;
this.modelInitializing = false;
if (this.sessionBackend === 'webgl') {
this.gpuSession = undefined;
} else {
this.cpuSession = undefined;
}
throw new Error('Error: Backend not supported. ');
}
this.modelLoading = false;async initSession() {
this.sessionRunning = false;
this.modelLoadingError = false;
if (this.sessionBackend === 'webgl') {
if (this.gpuSession) {
this.session = this.gpuSession;
return;
}
this.modelLoading = true;
this.modelInitializing = true;
this.gpuSession = new InferenceSession({backendHint: this.sessionBackend});
this.session = this.gpuSession;
}
if (this.sessionBackend === 'wasm') {
if (this.cpuSession) {
this.session = this.cpuSession;
return;
}
this.modelLoading = true;
this.modelInitializing = true;
this.cpuSession = new InferenceSession({backendHint: this.sessionBackend});
this.session = this.cpuSession;
}
try {
await this.session!.loadModel(this.modelFile);
} catch (e){async initSession() {
this.sessionRunning = false;
this.modelLoadingError = false;
if (this.sessionBackend === 'webgl') {
if (this.gpuSession) {
this.session = this.gpuSession;
return;
}
this.modelLoading = true;
this.modelInitializing = true;
this.gpuSession = new InferenceSession({backendHint: this.sessionBackend});
this.session = this.gpuSession;
}
if (this.sessionBackend === 'wasm') {
if (this.cpuSession) {
this.session = this.cpuSession;
return;
}
this.modelLoading = true;
this.modelInitializing = true;
this.cpuSession = new InferenceSession({backendHint: this.sessionBackend});
this.session = this.cpuSession;
}
try {
await this.session!.loadModel(this.modelFile);
} catch (e){async initSession() {
this.sessionRunning = false;
this.modelLoadingError = false;
if (this.sessionBackend === 'webgl') {
if (this.gpuSession) {
this.session = this.gpuSession;
return;
}
this.modelLoading = true;
this.modelInitializing = true;
this.gpuSession = new InferenceSession({backendHint: this.sessionBackend});
this.session = this.gpuSession;
}
if (this.sessionBackend === 'wasm') {
if (this.cpuSession) {
this.session = this.cpuSession;
return;
}
this.modelLoading = true;
this.modelInitializing = true;
this.cpuSession = new InferenceSession({backendHint: this.sessionBackend});
this.session = this.cpuSession;
}
try {
await this.session!.loadModel(this.modelFile);
} catch (e){this.session = this.gpuSession;
return;
}
this.modelLoading = true;
this.modelInitializing = true;
this.gpuSession = new InferenceSession({backendHint: this.sessionBackend});
this.session = this.gpuSession;
}
if (this.sessionBackend === 'wasm') {
if (this.cpuSession) {
this.session = this.cpuSession;
return;
}
this.modelLoading = true;
this.modelInitializing = true;
this.cpuSession = new InferenceSession({backendHint: this.sessionBackend});
this.session = this.cpuSession;
}
try {
await this.session!.loadModel(this.modelFile);
} catch (e){
this.modelLoading = false;
this.modelInitializing = false;
if (this.sessionBackend === 'webgl') {
this.gpuSession = undefined;
} else {
this.cpuSession = undefined;
}
throw new Error('Error: Backend not supported. ');
}
this.modelLoading = false;this.session = this.gpuSession;
return;
}
this.modelLoading = true;
this.modelInitializing = true;
this.gpuSession = new InferenceSession({backendHint: this.sessionBackend});
this.session = this.gpuSession;
}
if (this.sessionBackend === 'wasm') {
if (this.cpuSession) {
this.session = this.cpuSession;
return;
}
this.modelLoading = true;
this.modelInitializing = true;
this.cpuSession = new InferenceSession({backendHint: this.sessionBackend});
this.session = this.cpuSession;
}
try {
await this.session!.loadModel(this.modelFile);
} catch (e){
this.modelLoading = false;
this.modelInitializing = false;
if (this.sessionBackend === 'webgl') {
this.gpuSession = undefined;
} else {
this.cpuSession = undefined;
}
throw new Error('Error: Backend not supported. ');
}
this.modelLoading = false;onnxjs
A Javascript library for running ONNX models on browsers and on Node.js
