Android 如何更改TFlite模型输出的维度数？

我正在用谷歌的MLKit做一个Android应用程序来进行物体检测。 我知道有一些TFlite型号与MLKit兼容，但当我大部分时间使用其他型号时，我会出现以下错误：

 com.google.android.gms.tasks.RuntimeExecutionException: com.google.mlkit.common.MlKitException: Failed to initialize detector. Unexpected number of dimensions for output index 0: got 3D, expected either 2D (BxN with B=1) or 4D (BxHxWxN with B=1, W=1, H=1).

我想知道是否有可能（以及如何）操纵输入/输出（对于某些型号，问题是其他型号的输入）以满足MLKit的要求：

---

我在试着把一只方形猪放进一个圆孔里？

不要试着把方形猪[sic]放进一个圆孔里——这对方形猪来说是痛苦的。您可以选择一个圆形清管器，一个与ML套件兼容的型号，如下所示：