快速从 .npy 转换为 MLMultiArray 以进行 CoreML 预测

【问题标题】：Convert from .npy to MLMultiArray for CoreML prediction in swift快速从 .npy 转换为 MLMultiArray 以进行 CoreML 预测
【发布时间】：2020-09-03 08:25:06
【问题描述】：

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

【问题讨论】：

标签： swift numpy pytorch coreml

【解决方案1】：

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

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

现在您可以简单地将其加载到 Swift 中的 Data 对象中，然后将其复制到 MLMultiArray 中。

【讨论】：