如何在C ++中构造Tensorflow Lite的输入?

时间:2019-05-21 07:30:20

标签: c++ embedded-linux tensorflow-lite

我试图弄清楚应该如何为张量流精简模型寻找输入缓冲区。

我的输入应该是(1,224,224,3)图片缓冲区。

当我在答案上输入带有0或255(黑白)图像的输入缓冲区时,我会得到相同的答案。

uchar* in_data = new uchar[224*224*3];
for(int i=0; i<224*224*3;i++){
   // in_data[i] = 0;
   in_data[i] = 255;
}

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

input_1 = in_data;


此代码为我输入的所有数据提供了相同的答案。 当模型尺寸为(1,224,224,3)时,应如何构造适当的输入?

对于简单的情况,当我只有(1,128)个一维矢量时,一切工作正常。但是对于这种多维案例,我不知道如何进行。

0 个答案:

没有答案