我试图弄清楚应该如何为张量流精简模型寻找输入缓冲区。
我的输入应该是(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)个一维矢量时,一切工作正常。但是对于这种多维案例,我不知道如何进行。