c++ - 如何在 C++ 中为 tensorflow-lite 设置带图像的输入？

Question

我正尝试在嵌入式平台上将我们的 Tensoflow 模型从 Python+Keras 版本迁移到使用 C++ 的 Tensorflow Lite。

看来我不知道如何正确设置解释器的输入。

输入形状应为 (1, 224, 224, 3)。

作为输入，我使用 openCV 拍摄图像，并将其转换为 CV_BGR2RGB。

std::unique_ptr<tflite::FlatBufferModel> model_stage1 = 
tflite::FlatBufferModel::BuildFromFile("model1.tflite");
  TFLITE_MINIMAL_CHECK(model_stage1 != nullptr);

  // Build the interpreter
  tflite::ops::builtin::BuiltinOpResolver resolver_stage1;
  std::unique_ptr<Interpreter> interpreter_stage1;
  tflite::InterpreterBuilder(*model_stage1, resolver_stage1)(&interpreter_stage1);

TFLITE_MINIMAL_CHECK(interpreter_stage1 != nullptr);

  cv::Mat cvimg = cv::imread(imagefile);
  if(cvimg.data == NULL) {
    printf("=== IMAGE READ ERROR ===\n");
    return 0;
  }

  cv::cvtColor(cvimg, cvimg, CV_BGR2RGB);

  uchar* input_1 = interpreter_stage1->typed_input_tensor<uchar>(0);

 memcpy( ... );

我无法正确设置此 uchar 类型的 memcpy。

当我这样做时，我在工作期间出现段错误:

memcpy(input_1, cvimg.data, cvimg.total() * cvimg.elemSize());

在这种情况下，我应该如何正确填写输入？

Answer 1

要将我的评论转换为答案:Memcpy 在这里可能不是正确的方法。 OpenCV 将图像保存为每个像素的 RGB 排序(或 BGR 或另一种颜色组合)颜色值的一维数组。可以通过以下方式迭代这些 RGB block :

for (const auto& rgb : cvimg) {
    // now rgb[0] is the red value, rgb[1] green and rgb[2] blue.
}

并且将值写入 Tensorflow-Lite typed_input_tensor 应该像这样完成；其中 i 是索引(迭代器)，x 是分配的值:

interpreter->typed_input_tensor<uchar>(0)[i] = x;

所以循环看起来像这样:

for (size_t i = 0; size_t < cvimg.size(); ++i) {
    const auto& rgb = cvimg[i];
    interpreter->typed_input_tensor<uchar>(0)[3*i + 0] = rgb[0];
    interpreter->typed_input_tensor<uchar>(0)[3*i + 1] = rgb[1];
    interpreter->typed_input_tensor<uchar>(0)[3*i + 2] = rgb[2];
}