Настройка TensorFlow-Lite Swift (заголовок моста/включает/корневая папка)

Недавно был выпущен TensorFlow Lite. База кода поставляется с демонстрационным приложением для iOS.

Я хотел бы создать/запустить одно и то же приложение, но с помощью Swift. Я выполнил следующие шаги:

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

Этого недостаточно.

1) Я думаю, что сейчас мне не хватает какого-то соединительного заголовка.

2) Я также не уверен, как эти конкретные включения Obj-C будут выглядеть быстро:

#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+ МБ) быть включен в упакованное приложение ios?


person rambossa    schedule 17.11.2017    source источник


Ответы (1)


На данный момент вам нужно вызвать API TFLite C++ из Objective-C++. Вы можете создать тонкий класс Objective-C++ для вызова TFLite и использовать его из Swift.

В будущем может появиться больше языковой поддержки.

Для вопроса (3) скомпилированная библиотека будет состоять всего из нескольких сотен байт. Ознакомьтесь с библиотекой, созданной tensorflow/contrib/lite/build_ios_universal_lib.sh.

person miaout17    schedule 27.07.2018