从 MLModel 处理 MLMultiArray

Posted 2023-03-27

【中文标题】从 MLModel 处理 MLMultiArray

【英文标题】：Dealing with MLMultiArray from MLModel

【发布时间】：2019-10-06 15:16:03

【问题描述】：

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

【参考方案1】：

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

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