интерпретатор-> typed_input_tensor проблемы с переменными uchar - PullRequest
0 голосов
/ 24 мая 2019

У меня проблема с вводом заполнения модели tenorflow-lite.

Первый этап моей модели принимает переменные uchar, но:

//seg falut
    interpreter_stage1->typed_input_tensor<uchar>(0)[i] = 0;

//works good
    interpreter_stage1->typed_input_tensor<float>(0)[i] = 0;

Я не понимаю, почему я не могу понять, почему ucharВ этом случае тип не работает должным образом.

1 Ответ

0 голосов
/ 24 мая 2019

Вот ответ, если кто-то столкнется с похожей проблемой.

Когда я печатал interpretetState

tflite::PrintInterpreterState(interpreter.get());

Я видел, что модель, которая ранее была перемещена из Keras в TFLite, имеет тип ввода, отличный от ожидаемого. Похоже, что он ждет Float32 не учар типа.

Interpreter has 26 tensors and 11 nodes
Inputs: 9
Outputs: 16

Tensor   0 conv2d_1/BiasAdd     kTfLiteFloat32  kTfLiteArenaRw    3154176 bytes ( 3.0 MB)  1 222 222 16
(...)
...