onnxruntime-web

What is onnxruntime-web?

onnxruntime-web is a JavaScript library that allows you to run ONNX (Open Neural Network Exchange) models directly in the browser or in Node.js environments. It leverages WebAssembly (WASM) and WebGL for efficient execution of machine learning models, making it possible to perform inference tasks on the client side without needing a server.

What are onnxruntime-web's main functionalities?

Load and Run ONNX Model

This feature allows you to load an ONNX model and run inference on it. The code sample demonstrates how to create an inference session, prepare input data, and run the model to get the results.

const ort = require('onnxruntime-web');

async function runModel() {
  const session = await ort.InferenceSession.create('model.onnx');
  const input = new ort.Tensor('float32', new Float32Array([1.0, 2.0, 3.0, 4.0]), [1, 4]);
  const feeds = { input: input };
  const results = await session.run(feeds);
  console.log(results);
}

runModel();

Use WebGL Backend

This feature allows you to leverage the WebGL backend for running ONNX models, which can provide better performance for certain types of models. The code sample shows how to specify the WebGL execution provider when creating the inference session.

const ort = require('onnxruntime-web');

async function runModelWithWebGL() {
  const session = await ort.InferenceSession.create('model.onnx', { executionProviders: ['webgl'] });
  const input = new ort.Tensor('float32', new Float32Array([1.0, 2.0, 3.0, 4.0]), [1, 4]);
  const feeds = { input: input };
  const results = await session.run(feeds);
  console.log(results);
}

runModelWithWebGL();

Run Model in Node.js

This feature demonstrates how to run ONNX models in a Node.js environment. The code sample is similar to the browser example but is intended to be run in a Node.js context.

const ort = require('onnxruntime-web');

async function runModelInNode() {
  const session = await ort.InferenceSession.create('model.onnx');
  const input = new ort.Tensor('float32', new Float32Array([1.0, 2.0, 3.0, 4.0]), [1, 4]);
  const feeds = { input: input };
  const results = await session.run(feeds);
  console.log(results);
}

runModelInNode();

Other packages similar to onnxruntime-web

brain.js

Brain.js is a JavaScript library for neural networks, which can be used in the browser or with Node.js. It is simpler and more lightweight compared to onnxruntime-web, and is suitable for basic neural network tasks. However, it does not support the ONNX model format.

synaptic

Synaptic is a JavaScript neural network library for node.js and the browser. It provides a flexible and powerful API for creating and training neural networks. Unlike onnxruntime-web, Synaptic does not support running pre-trained ONNX models and is more focused on building and training models from scratch.

README

The README file is missing