react-native-tensorflow-lite

Tensorflow Lite for React Native (Currently iOS only).

Installation

npm install @switt/react-native-tensorflow-lite

Current features

• Documented in Typescript. Seriously, just read the types from there.
• Allow selection of model simply by local file URL (no need to add to compile-time). Was designed to be used with Expo's Asset system (resolve model file URI from Asset.fromModule(...))
• Input type : Image(s)
  • Input 1 or multiple images to be run on a Tensorflow Lite model.
  • Allow selecting resize mode of the images (fit vs fill)
  • Allow specifying crop areas (especially useful when using a previous model to crop out a region of interest)
• iOS :: TensorflowLite with TF Ops enabled (version 2.6.0 - 0.0.1 nightly fails linking due to duplicated symbols. Will update if stable.)
  • simply pod install and the linker flags are added automatically.

Usage

import TensorflowLite from "@switt/react-native-tensorflow-lite";

// ...
const imageUris : string[] = ['local_path_to_image_1', 'local_path_to_image_2']

const modelAsset = Asset.fromModule(require('./face_landmark.tflite'));
await modelAsset.downloadAsync();

// Operation is run in batches of files. This is to minimize the number of react native bridge calls.
const results = await TensorflowLite.runModelWithFiles({
  model: modelAsset.localUri!,
  files: imageUris
});
this.setState({ results });

Contributing

See the contributing guide to learn how to contribute to the repository and the development workflow.

License

MIT