Skip to content

Files

Latest commit

 

History

History
165 lines (116 loc) · 4.83 KB

README_CN.md

File metadata and controls

165 lines (116 loc) · 4.83 KB

LibTensorFlowForiOSSwift

English Document

这是一个TensorFlow运行在iOS上的demo，使用swift开发的实例，可以从用户输入的文本预测用户可能需要输入的emoji表情。

在 emoji-tf-ios基础上进行修改；使用了emoji-tf-ios 的 emoji_frozen.pb 模型。

如何运行

  • 打开终端进入项目的根目录

  • 输入 sh run.sh 运行

最后会自动的打开Xcode，此时应该可以运行项目

关于 run.sh

run.sh会自动的编译 TensorFlow for iOS

关于项目配置

如何在iOS项目中导入TensorFlow

  • 下载TensorFlow到项目根目录并编译

    在编译源文件之前, 要先进行修改TensorFlow Kernel的一些文件（版本 <= 1.2.1）

    在 run.sh 中会自动去修改这些文件

    kernel path: tensorflow/tensorflow/core/kernels

    • cwise_op_add_1.cc

      源码:

       ...
 ...
 #include "tensorflow/core/kernels/cwise_ops_common.h"

 namespace tensorflow {
 REGISTER5(BinaryOp, CPU, "Add", functor::add, float, Eigen::half, double, int32,
           int64);

 #if TENSORFLOW_USE_SYCL
 ...
 ...

      修改为:

       ...
 ...
 #include "tensorflow/core/kernels/cwise_ops_common.h"

 namespace tensorflow {
 REGISTER5(BinaryOp, CPU, "Add", functor::add, float, Eigen::half, double, int32,
           int64);

 // line 21 insert this code
 #if defined(__ANDROID_TYPES_SLIM__)
 REGISTER(BinaryOp, CPU, "Add", functor::add, int32);
 #endif  // __ANDROID_TYPES_SLIM__
 // insert end

 #if TENSORFLOW_USE_SYCL
 ...
 ...

    • cwise_op_less.cc

      源码:

       ...
 ...
 #include "tensorflow/core/kernels/cwise_ops_common.h"

 namespace tensorflow {
 REGISTER8(BinaryOp, CPU, "Less", functor::less, float, Eigen::half, double,
           int32, int64, uint8, int8, int16);
 #if GOOGLE_CUDA
 REGISTER7(BinaryOp, GPU, "Less", functor::less, float, Eigen::half, double,
           int64, uint8, int8, int16);
 ...
 ...

      修改为:

       ...
 ...
 #include "tensorflow/core/kernels/cwise_ops_common.h"

 namespace tensorflow {
 REGISTER8(BinaryOp, CPU, "Less", functor::less, float, Eigen::half, double,
           int32, int64, uint8, int8, int16);
 
 // line 21 insert this code
 #if defined(__ANDROID_TYPES_SLIM__)
 REGISTER(BinaryOp, CPU, "Less", functor::less, int32);
 #endif  // __ANDROID_TYPES_SLIM__
 // insert end

 #if GOOGLE_CUDA
 REGISTER7(BinaryOp, GPU, "Less", functor::less, float, Eigen::half, double,
           int64, uint8, int8, int16);
 ...
 ...

  • libtensorflow-core.a

    • Other Link Flags 中加入 $(SRCROOT)/tensorflow/tensorflow/contrib/makefile/gen/lib/libtensorflow-core.a
    • Library Search Paths 中加入 $(SRCROOT)/tensorflow/tensorflow/contrib/makefile/gen/lib

  • libprotobuf.a & libprotobuf-lite.a

    • Build Phases | Link Binary With Libraries 中加入 libprotobuf.a & libprotobuf-lite.a (path: tensorflow/tensorflow/contrib/makefile/gen/protobuf_ios/lib/)
    • Library Search Paths 中加入 $(SRCROOT)/tensorflow/tensorflow/contrib/makefile/gen/protobuf_ios/lib

  • Header Search Paths中加入

    • $(SRCROOT)/tensorflow/
    • $(SRCROOT)/tensorflow/tensorflow/contrib/makefile/downloads/protobuf/src/
    • $(SRCROOT)/tensorflow/tensorflow/contrib/makefile/downloads
    • $(SRCROOT)/tensorflow/tensorflow/contrib/makefile/downloads/eigen
    • $(SRCROOT)/tensorflow/tensorflow/contrib/makefile/gen/proto

  • Other Link Flags 中加入 -force_load

  • Build Phases | Link Binary With Libraries 中加入 Accelerate.framework

  • C++ Language Dialect 设置为 GNU++11 or GNU++14

  • C++ Standard Library 设置为 libc++

  • Enable Bitcode 设置为 No

  • 删除所有使用的 -all_load ，替换为 -ObjC

    Remove any use of the -all_load flag in your project. The protocol buffers libraries (full and lite versions) contain duplicate symbols, and the -all_load flag will cause these duplicates to become link errors. If you were using -all_load to avoid issues with Objective-C categories in static libraries, you may be able to replace it with the -ObjC flag.

  • 忽略TensorFlow编译的警告：

    • 在Other C Flags & Other C++ Flags中加入-isystem $(SRCROOT)/tensorflow

参考：