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

Posted 2023-03-26

【中文标题】快速从 .npy 转换为 MLMultiArray 以进行 CoreML 预测

【英文标题】：Convert from .npy to MLMultiArray for CoreML prediction in swift

【发布时间】：2020-09-03 08:25:06

【问题描述】：

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

【参考方案1】：

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

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

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