ios - TensorFlow-Lite Swift 设置(bridge-header/includes/root 文件夹)

Question

TensorFlow Lite 最近发布了。代码库带有 demo ios application .

我想创建/运行相同的应用程序，但使用 Swift。我已按照以下步骤操作:

https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/lite/g3doc/ios.md#using-in-your-own-application

这还不够。

1) 我想我现在缺少的是一些桥接头。

2) 我也不确定这些特定的 Obj-C 包含在 swift 中会是什么样子:

#include "tensorflow/contrib/lite/kernels/register.h"
#include "tensorflow/contrib/lite/model.h"
#include "tensorflow/contrib/lite/string_util.h"
#include "tensorflow/contrib/lite/tools/mutable_op_resolver.h"

3) TensorFlow 根目录(200+mb)是否必须包含在打包的 ios 应用程序中？

Answer 1

目前，您需要从 Objective-C++ 调用 TFLite C++ API。您可以创建一个瘦 Objective-C++ 类来调用 TFLite，并在 Swift 中使用它。

将来可能会提供更多语言支持。

对于问题(3)，编译后的库也就几百个字节。查看由 tensorflow/contrib/lite/build_ios_universal_lib.sh 构建的库。