tenorflow lite(tflite) вызывает ошибку после изменения размера входного демона

Question

Я использую mobilenet_ssd.tflite в качестве режима от официального тензор потока GitHub. Код ниже:

int input = interpreter->inputs()[0];
interpreter->ResizeInputTensor(input, sizes);

Это вызовет ошибку при вызове:

interpreter->AllocateTensors()

Если я закомментирую интерпретатор->ResizeInputTensor(ввод, размеры); То все в порядке.

Какие-либо предложения?

Еще один вопрос, который я задал: изменить размер входного изображения для mobilenet_ssd с помощью tenorflow

Answer 1

ResizeInputTensor ограничено архитектурой нейронной сети. Сбой, так как SSD MobileNet и MobileNet могут обрабатывать только ввод фиксированного размера.

То, что может работать, - это изменение размера партии. Например, вы можете попробовать изменить размер с (1, 244, 244, 3) на (4, 244, 244, 3) и выполнить вывод на 4 изображения в одном Invoke вызов.