如何使用CoreML模型输出MLMultiArray

Question

我需要解析 CoreML 模型的预测输出结果。

我可以看到输出类型是 MLMultiArray，但是当我在我的代码中使用这样的类型时：

let a = MultiArray<Float>(result.transpose_1_tmp_0)

总是报错：

Cannot find 'MultiArray' in scope

有人知道吗？

Answer 1

MultiArray 不是 built-in 数据类型。使用变量名：

var result: MLMultiArray //Your populated array
let a = result[[z, y, x] as [NSNumber]].floatValue

您可能还想查看 Swiftier MultiArray. Follow the CoreMLHelpers 安装说明。

编辑：

MLMultiArray 是 CoreML 框架的一部分。要使用它，您必须像这样导入框架：

import CoreML

How to use CoreML model output MLMultiArray