如何读取CoreML模型预测MLMultiArray

Question

我是新的CoreML，我已经生成了一个coreML模型，返回双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的每一行以及每一行的列的值。我在网上搜索了很多，但是没有找到任何可以在MLMultiArray上迭代的循环方法。任何解决办法。

Answer 1

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