从 MLModel 处理 MLMultiArray

Dealing with MLMultiArray from MLModel

我在我的项目中使用 mlmodel 来检测对象。模型的输入为 Image (Color 300 x 300),输出为 MultiArray (Double)MultiArray (Double 4 x 1917 x 1)。我做了 VNCoreMLRequest 并且我得到的输出与上面提到的完全一样。在这里,我需要从 MLMultiArray 输出中向用户显示对象名称和置信度。现在,问题是如何对其进行解码,如何从 MLMultiArray 中获取对象名称和置信度。我研究了过去 3 天,但没有用。

数字 1917 告诉我你最有可能使用 MobileNet + SSD。

我写了很多博客 post 来解释这种事情是如何工作的。这里有一篇博客 post 解释了如何对 MobileNetV2+SSDLite 进行解码,但同样的逻辑也适用于其他版本的 MobileNet 和 SSD:https://machinethink.net/blog/mobilenet-ssdlite-coreml/