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构建的库。

TensorFlow-Lite Swift设置（bridge-header / includes / root文件夹）

1 个答案: