如何设置输入与在C ++ tensorflow-精简版的形象？(How to set input w

我试图用Python + Keras版本我们Tensoflow模型搬到Tensorflow精简版与C ++的嵌入式平台上。

它看起来像我不知道如何正确设置输入翻译。

输入形状应该是（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());

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

要转换我的意见为回答：的memcpy可能不会来这里是正确的做法。 OpenCV的将图像保存为RGB有序的一维阵列（或BGR或另一颜色组合）每个像素的颜色值。它可以遍历通过这些RGB-块：

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

和值写入到Tensorflow - 精简版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];
}