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...etc)不支持可变输入形状。

如果您确定该模型应该支持变量输入形状，并且它不工作。请在此提交一个Github问题或回复。