Как построить вход для tenorflow lite в c ++? - PullRequest
0 голосов
/ 21 мая 2019

Я пытаюсь выяснить, как должен выглядеть входной буфер для модели tenorflow 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) одномерный вектор, все работает хорошо,Но в этом многомерном случае я не знаю, как поступить.

...