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

Question

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

Answer 1

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

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

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