apple/coremltools

Core ML tools contain supporting tools for Core ML model conversion, editing, and validation.

解决的问题

Core ML Tools 提供了一种将流行第三方库中训练的机器学习模型转换为 Core ML 格式的方法。这使得开发者能够将这些模型集成到 Apple 应用中,实现无需网络连接的设备端预测和微调,从而提升隐私保护和应用响应速度。

工作原理

这是一个作为转换器和优化器的 Python 包。它支持从 TensorFlow(1.x 和 2.x)、PyTorch、scikit-learn、XGBoost 和 LibSVM 等框架转换模型。转换后,模型将以 Core ML 格式保存,然后可通过 Xcode 集成到应用程序中。

适用人群

希望利用 Apple Core ML 框架所需相同模型格式,在 Apple 设备上部署机器学习模型的开发者。

主要亮点

  • 支持从广泛的神经网络和非神经网络框架转换模型。
  • 支持读取、写入和优化 Core ML 模型。
  • 可在 macOS 上进行预测,验证转换结果。
  • 支持通过 Apple 设备的 CPU、GPU 和 Neural Engine 实现硬件加速。

相关

  • 项目
  • Dispatch
  • 项目
  • 项目
  • 项目