tensorflow - tflite : tflite 有什么方法可以处理各种形状的输入

Question

我想在 tflite 中使用MTCNN 进行人脸检测，它应该适用于可变输入形状，例如

both 1920*1080 or 640*360

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

Answer 1

如果模型架构支持可变输入形状，您应该能够调用resizeInputAPI (Java) 或ResizeInputTensorAPI（如果您使用 C++）。

例如，在 Java 中：

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

请注意，许多网络架构（例如 MobileNet、Inception、ResNet...等）不支持可变输入形状。

如果您确定模型应该支持可变输入形状并且它不起作用。请提交 Github 问题或在此处回复。