iOS 应用中的 TF Mobile

TensorFlow 通过以下步骤支持 iOS 应用:

  1. 通过在项目的根目录中添加名为Profile的文件,在您的应用中包含 TF Mobile。将以下内容添加到Profile
target 'Name-Of-Your-Project'
       pod 'TensorFlow-experimental'
  1. 运行pod install命令下载并安装 TensorFlow 实验舱。
  2. 运行myproject.xcworkspace命令打开工作区,以便将预测代码添加到应用逻辑中。

To create your own TensorFlow binaries for iOS projects, follow the instructions at this link: https://github.com/tensorflow/tensorflow/tree/master/tensorflow/examples/ios

在 iOS 项目中配置 TF 库后,可以通过以下四个步骤调用 TF 模型:

  1. 加载模型:
PortableReadFileToProto(file_path, &tensorflow_graph);
  1. 创建会话:
tensorflow::Status s = session->Create(tensorflow_graph);
  1. 运行预测或推理并获得输出:
std::string input_layer = "input";
std::string output_layer = "output";
std::vector<tensorflow::Tensor> outputs;
tensorflow::Status run_status = session->Run(
    {{input_layer, image_tensor}},
 {output_layer}, {}, &outputs);
  1. 获取输出数据:
tensorflow::Tensor* output = &outputs[0];

results matching ""

    No results matching ""