如何阅读 CoreML model 预测 MLMultiArray

Question

I am new to CoreML and I have generated a coreML model that return the Double 9 × 4 matrix matrix as shown in image blow.

Here is the code that is working fine and giving me the 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

Now I have to read each row of the MLMultiArray and also the values of column of each row. I have search a lot on inte.net but did not find any looping method that can iterate on MLMultiArray. Any solution for this.

Answer 1

Have a look at https://github.com/pytorch/ios-demo-app/tree/master/ObjectDetection this helped me to understand aa bunch of stuff, also search for some yolo coreML