从 .npy 转换为 MLMultiArray 以在 swift 中进行 CoreML 预测

Question

我已经将 PyTorch 模型导出到 CoreML 并想在 swift 中进行推理。我将输入数据作为 2D float32 numpy ndarray .npy 存储在磁盘上，需要加载到 swift 中的 MLMultiArray 中。有没有方便的方法来做到这一点？

Answer 1

不保存为 .npy（已腌制），而是保存 NumPy 中的原始数据：

array.astype(np.float32).tofile(filename)

现在您只需将其加载到 Swift 中的数据对象中，然后将其复制到 MLMultiArray 中。