从MLModel处理MLMultiArray

时间:2019-05-20 09:16:12

标签: ios swift object-detection coreml mlmodel

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

1 个答案:

答案 0 :(得分:1)

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

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