Configuration de TensorFlow-Lite Swift (dossier bridge-header/includes/Root)

Question

TensorFlow Lite a été publié récemment. La base de code est fournie avec un application ios de démonstration .

Je voudrais créer/exécuter la même application, mais en utilisant Swift. J'ai suivi les étapes suivantes :

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

Ce n'est pas suffisant.

1) Je pense que ce qui me manque maintenant, c'est un en-tête de pontage.

2) Je ne sais pas non plus comment ces inclusions Obj-C spécifiques se présenteraient dans 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) Le répertoire racine de TensorFlow (200+mb) doit-il être inclus dans l'application ios packagée ?

Answer 1

Pour l'instant, vous devez appeler l'API C++ de TFLite depuis Objective-C++. Vous pouvez créer une classe Objective-C++ fine pour appeler TFLite, et l'utiliser depuis Swift.

D'autres langues pourraient être prises en charge à l'avenir.

Pour la question (3), la bibliothèque compilée ne serait que de quelques centaines d'octets. Regardez la bibliothèque construite par tensorflow/contrib/lite/build_ios_universal_lib.sh .