tflite：有没有办法让tflite处理各种形状的输入

Question

我想在tflite中使用 MTCNN 进行面部检测，它应该适用于像

这样的可变输入形状

  

both 1920*1080 or 640*360

但是tflite必须将输入形状设置为某个值，有没有办法让tflite接受变量输入或输出形状？ 非常感谢

Answer 1

如果模型体系结构支持可变的输入形状，则应该能够调用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问题或在此处回复。