margelo/react-native-fast-tflite

🧬 High-performance TensorFlow Lite library for React Native with GPU acceleration

解决的问题

react-native-fast-tflite 是一个高性能库,允许 React Native 开发者在移动设备上运行 TensorFlow Lite (.tflite) 模型。它弥合了 JavaScript 与低级 C/C++ TensorFlow Lite 核心 API 之间的差距,使无需传统 React Native 桥接开销即可实现高效的设备端 AI 推理。

工作原理

该库使用 Nitro Modules 实现零拷贝 ArrayBuffers,意味着数据在原生端与 JavaScript 之间传递时无需昂贵的复制操作。它直接与 TensorFlow Lite 核心 API 交互,实现直接内存访问。模型可从应用包、本地文件或远程 URL 加载,并可在运行时动态切换。

适用人群

需要在 iOS 和 Android 上集成机器学习模型以完成图像分类、目标检测或其他基于张量的预测任务的 React Native 移动开发者。

主要亮点

  • 高性能:基于 Nitro Modules 实现零拷贝数据传输。
  • GPU 加速:支持包括 CoreML(iOS)、Metal、OpenGL 以及 Android GPU/NNAPI 在内的 GPU 加速代理。
  • 灵活的模型加载:可通过 require、本地 URI 或远程 URL 加载模型。
  • 与 VisionCamera 集成:专为与 react-native-vision-camera 轻松集成而设计,适用于实时 AI 视觉任务。

相关

  • 项目
  • 项目
  • 项目
  • 项目