Tensorflow.js Onnx Runner

Run and finetune pretrained Onnx models in the browser with GPU support via the wonderful [Tensorflow.js][tfjs] library.

Usage

$3

You can use this as standalone es5 bundle like this:

``html

`

Then, loading model is a simple as referencing the path to the model.onnx file.

Here is an example of loading GoogLeNet:

`js
var modelUrl = 'models/bvlc_googlenet/model.onnx';

// Initialize the tf.model
var model = new tf.onnx.loadModel(modelUrl);

// Now use tf.model
const pixels = tf.fromPixels(img);
const predictions = model.predict(pixels);
`

$3

To run the demo, use the following:

`bash
npm run build

Start a webserver


npm run start
`

Now navigate to http://localhost:8080/demos.

> Hint: some of the models are quite big (>30MB). You have to download the Onnx models and place them into the demos/models directory to save bandwith.

Development

`sh
npm install
`

To build a standalone bundle run

`sh
npm run build
``

[tfjs]: https://github.com/tensorflow/tfjs