我的 coremlmodel 在 swift 上给出了不同的输出答案

【问题标题】：My coremlmodel gives different output on swift我的 coremlmodel 在 swift 上给出了不同的输出
【发布时间】：2021-04-30 01:28:25
【问题描述】：

我像这样创建我的 MLMultiArray

var mlInput = try? MLMultiArray(shape: [1,2,12,120], dataType: MLMultiArrayDataType.float32)

并用我从下面的 json 文件中获得的值填充它

mlInput![0 * 12 * 120 + 0 * 120 + 5] = value as! NSNumber

基本上我将索引 0 置于 2 之上，索引 0 置于 12 之上，索引 5 置于 120 之上

并将其提供给如下模型

let obj = try! mlModel.prediction(input: ml3dmodelInput(input: mlInput))

并通过您看到的这些步骤获得结果。它给出了错误的预测输出。我在 python 中做了同样的事情来测试 coremlmodel 它在 python 上运行良好。我只是做了相同的步骤，用来自 json 的值填充数组，将其提供给模型并获取输出。我做错了什么？

【问题讨论】：

标签： ios swift coreml

【解决方案1】：

您应该使用数组的strides 来计算索引。

如果数组的形状为[A, B, C, D]，那么正确的索引是：

a * strides[0] + b * strides[1] + c * strides[2] + d * strides[3]

不过，您也可以这样做：

mlInput[[a, b, c, d] as [NSNumber]] = value

【讨论】：

我知道这是保证方法，但索引没有错。我还尝试了您建议的索引方法中的第二个，但效果不佳。

【解决方案2】：

仅使用 cpu 解决了问题。

let options = MLPredictionOptions()
options.usesCPUOnly = true
let obj = try! mlModel.prediction(input: ml3dmodelInput(input: mlInput), options: options)

像这样改变了第三步。

【讨论】：