如何解决我的 coremlmodel 在 swift
我像这样创建我的 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 的值填充数组,将其提供给模型并获得输出。我做错了什么?
解决方法
您应该使用数组的 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
,
仅使用cpu解决了问题。
let options = MLPredictionOptions()
options.usesCPUOnly = true
let obj = try! mlModel.prediction(input: ml3dmodelInput(input: mlInput),options: options)
第三步改成这样。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。