用cpp推断tflite模型

Question

我试着用cpp推断我的tflite模型。我已经训练了模型并保存了h5文件。现在我试着用cpp加载它并得到预测。我试过以下一组代码。

#include <iostream>
#include <cstdio>
#include "tensorflow/lite/interpreter.h"
#include "tensorflow/lite/kernels/register.h"
#include "tensorflow/lite/model.h"
#include "tensorflow/lite/tools/gen_op_registration.h"

int main(int argc, char** argv) {
    auto model = tflite::FlatBufferModel::BuildFromFile("model.tflite");   //CHANGED

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);
    
    return 0;
}

但是我得到了一个错误，因为：

3   41  E:\kaino\TF\tf\app\main.cpp [Error] tensorflow/lite/interpreter.h: No such file or directory

我是否必须安装一个软件包，还是必须添加另一个文件？我是新来的，有人能帮我解决这个问题吗？

Answer 1

您可能丢失了编译器的包含目录。基本上，您希望包含“tensorflow/lite/.”等所有文件的路径。是否存在，以便编译器可以搜索此目录并找到所需的头文件。

比如GCC，

g++ main.cc -I /tmp/my_local_path

有关更多信息，https://gcc.gnu.org/onlinedocs/gcc/Directory-Options.html#Directory-Options