Google MLkit 具有图像分类模型的对象检测边界框

Question

因此我使用 google MlKit 及其对象检测 API 开发了一个应用程序。我注意到的一件事是，我使用的 tflite 图像分类模型（ TFLite (efficientnet/lite3/int8) ）仅输出对象可能属于的类别的概率，这与大多数也返回边界框值的对象检测模型不同。然而，google mlkits 对象检测 API 仅支持图像分类模型。然而，API 也以某种方式返回给我装箱框值。我的问题是，当 tflite 模型本身仅返回类概率列表时，api 如何返回边界框值。

然而，API 也以某种方式返回给我装箱框值。我的问题是，当 tflite 模型本身仅返回类概率列表时，api 如何返回边界框值。

Answer 1

ML Kit 对象检测 SDK 管道包含两个模型：检测图像中对象的检测器模型，以及在检测到的对象的裁剪图像上运行的分类模型。

探测器型号是内部的，您无法更改。

您可以通过 API 的自定义变体更改的部分是分类模型。