如何使用Coreml模型输出MLMultiarRay

Question

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

我可以看到输出的类型是 MLMultiArray，但是当我在代码中使用该类型时，如下所示：

let a = MultiArray(result.transpose_1_tmp_0)

它总是抛出错误：

在范围内找不到“MultiArray”

有人知道吗？

Answer 1

Multiarray不是内置数据类型。使用变量名：

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

您也可能需要查看 a>。请按照 coremlhelpers 安装指令。

编辑：

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

import CoreML