如何阅读 CoreML model 预测 MLMultiArray

Question

我是 CoreML 的新手，我生成了一个 coreML model，它返回 Double 9 × 4 矩阵，如下图所示。

这是工作正常并为我提供 MLMultiArray 的代码。

 let outFeatures = try model.prediction(from: input)
        
        // get all 4 featrue array from image
        
        // MLFeature Value
        //1 × 3 × 20 × 20 × 8
        let confidence = outFeatures.featureValue(for: "confidence")?.multiArrayValue
        let coordinates = outFeatures.featureValue(for: "coordinates")?.multiArrayValue

现在我必须读取 MLMultiArray 的每一行以及每一行的列值。 我在 inte.net 上搜索了很多，但没有找到任何可以在 MLMultiArray 上迭代的循环方法。 任何解决方案。

Answer 1

看看https://github.com/pytorch/ios-demo-app/tree/master/ObjectDetection这帮助我理解了一堆东西，还搜索了一些 yolo coreML