node-tflite is an unofficial TensorFlow Lite 2.2 bindings for Node.js.
It can run TensorFlow Lite models (.tflite
) in Node.js environment (including Electron).
.tflite
) while TensorFlow.js cannot (you'll need a different way of model conversion in TensorFlow.js)npm install node-tflite
import { Interpreter } from "node-tflite";
const modelData = fs.readFileSync("/path/to/model.tflite");
const interpreter = new Interpreter(modelData);
interpreter.allocateTensors();
interpreter.inputs[0].copyFrom(inputData);
interpreter.invoke();
interpreter.outputs[0].copyTo(outputData);
TODO
npm install
npm test
npm run dist
bazel build //tensorflow/lite/c:tensorflowlite_c