Description:

tflite web

TFLite Web #
Run TFLite models on Dart JS. It is packaged in a WebAssembly binary that runs in a browser
Getting Started #


Unpack tflite (link) into your web folder.
Result structure:
-- Web
├── tflite
──────├── tf-backend-cpu.js
──────├── tf-core.js
──────├── tf-tflite.min.js
──────├── tflite_web_...


Initialize TFLite:


await TFLiteWeb.initialize();
copied to clipboard
This may take a couple of seconds

Either Load Model from Url:

final loadedModel = await TFLiteModel.fromUrl(modelUrl);
copied to clipboard

Or load from memory:

final loadedModel = await TFLiteModel.fromMemory(modelBytes);
copied to clipboard

Create a tensor:

final input = createTensor(data, shape, dataType);
copied to clipboard

Run the model:

final outputs = loadedModel.predict(input);
copied to clipboard
Depending on the model, the input can be a Tensor, a list of Tensors, or a Tensor map
Current Version: #

TF-JS: 4.11.0
TF-JS TFLite: 0.0.1-alpha.10

License

For personal and professional use. You cannot resell or redistribute these repositories in their original state.

Share

• Share on Facebook
• Share on Twitter