tflite: есть ли способ для tflite обрабатывать различные формы ввода

Я хочу использовать MTCNN для обнаружения лиц в tflite, он должен быть доступен для переменной формы ввода, например

both 1920*1080 or 640*360

но tflite должен установить форму ввода на определенное значение, есть ли способ позволить tflite принимать переменную форму ввода или вывода? огромное спасибо


tensorflow tensorflow-lite
person Lxf    schedule 30.05.2018    source источник

Ответы (1)


arrow_upward
1
arrow_downward

Если архитектура модели поддерживает переменную форму ввода, вы должны иметь возможность вызывать resizeInput API (Java) или ResizeInputTensor API (если вы используете C++).

Например, в Java:

interpreter.resizeInput(tensor_index, [1, 1920, 1080, 3]);
interpreter.resizeInput(tensor_index, [1, 640, 360, 3]);

Обратите внимание, что многие сетевые архитектуры (например, MobileNet, Inception, ResNet и т. д.) не поддерживают переменную форму ввода.

Если вы уверены, что модель должна поддерживать переменную форму ввода, и она не работает. Пожалуйста, сообщите о проблеме Github или ответьте здесь.

person miaout17    schedule 08.02.2019