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

Question

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

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

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

interpreter->AllocateTensors()

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

Есть предложения?

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

Answer 1

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

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