如何在C ++项目中使用TF Lite库

时间:2019-06-13 04:34:59

标签: c++ tensorflow tensorflow-lite

过去1-2天,我一直在努力构建TensorFlow Lite,因此我可以在自己的C \ C ++项目中将其用作标头或库。

例如,我有一个main.cpp的C ++项目,其代码如下:

#include "tensorflow/lite/model.h"
#include "tensorflow/lite/interpreter.h"
#include "tensorflow/lite/kernels/register.h"

int main()
{
    std::unique_ptr<tflite::FlatBufferModel> model;
    model = tflite::FlatBufferModel::BuildFromBuffer(h5_converted_tflite, h5_converted_tflite_len);

    tflite::ops::builtin::BuiltinOpResolver resolver;
    std::unique_ptr<tflite::Interpreter> interpreter;
    tflite::InterpreterBuilder(*model, resolver)(&interpreter);

    // Resize input tensors, if desired.
    interpreter->AllocateTensors();

    float* input = interpreter->typed_input_tensor<float>(0);
    // Fill `input`.

    interpreter->Invoke();

    float* output = interpreter->typed_output_tensor<float>(0);
}

我应该从哪里下载\ build文件,然后从哪里下载,以便我可以成功地编译此代码?当前,它显然说找不到h文件,并且当我克隆TF存储库并将其添加到include文件夹时,它找不到“ flatbuffers.h”文件,而当我手动添加它时,它给出了我有很多联系错误。 任何帮助将不胜感激...

预先感谢

2 个答案:

答案 0 :(得分:0)

请尝试以下经过variable value cont 27 cont 83 cont 43 cont 45 treat 49040 treat 53522 treat 48570 treat 22235 . . . lite 1.14.0测试的代码:

TensorFlow

答案 1 :(得分:0)

最近,TFLite 支持 CMake,它似乎解决了您的依赖问题。

https://www.tensorflow.org/lite/guide/build_cmake#create_a_cmake_project_which_uses_tensorflow_lite