运行 android 中的张量流推理图

Running tensoflow inference graph in android

我是应用程序开发新手。我想 运行 在 android 中训练我的对象检测模型。该模型由Object Detection API制作。因此输出为 frozen_inference_graph.pb 格式。

我查看了文档 page。它提供了多种选择。我不知道哪一个适合我。关于在以下路径中选择哪条路径的任何建议:

  1. 使用来自 JCenter 的 TensorFlow AAR 在 Android Studio 中构建
  2. 使用来自源代码的 TensorFlow 构建演示

感谢和问候 skbhat

他们各有利弊,但无论哪种方式都可以完成工作。

  1. 使用 jni 包装器获取预构建库 (tensorflow_inference.so) 使您能够轻松构建应用程序,因此我认为这是推荐的开始方式。

  2. 从源代码构建需要一些时间,有时还很麻烦(取决于您当前的环境),但它为您提供了高度 customize/optimize 的整体功能。所以我想如果你有足够的知识来操作核心 api(比如 adding/removing ops 等)

  3. 会更好

简而言之,在我看来,如果您是 android tensorflow 的新手,那么从 JCenter 获取库就足够了。