将机器学习模型转换为onnx,以便在windows ml中使用

winmltools的Python项目详细描述


winmltools为windows ml提供以下工具:

模型转换

winmltools允许您从不同的计算机转换模型 在ONNX中学习工具包,以便与windows ml一起使用。 目前支持以下工具包:

  • 苹果coreml
  • 路缘石
  • SCIKIT学习
  • lightgbm
  • xgboost
  • libsvm
  • SPARKML(实验)
  • 张量流(实验)

下面是一个转换核心ml模型的简单示例:

from coremltools.models.utils import load_spec
from winmltools import convert_coreml
model_coreml = load_spec('example.mlmodel')
model_onnx = convert_coreml(model_coreml, 7, name='ExampleModel')

训练后权重量化

winmltools提供量化工具来减少模型的内存占用。

下面是将onnx模型转换为量化onnx模型的示例:

import winmltools

model = winmltools.load_model('model.onnx')
quantized_model = winmltools.quantize(model, per_channel=True, nbits=8, use_dequantize_linear=True)
winmltools.save_model(quantized_model, 'quantized.onnx')

依赖性

要从不同的工具包进行转换,可能需要为不同的转换器安装以下软件包:

ToolkitSource
kerashttps://pypi.org/project/Keras
tensorflowhttps://pypi.org/project/tensorflow
scikit-learnhttps://pypi.org/project/scikit-learn
lightgbmhttps://pypi.org/project/lightgbm
xgboosthttps://pypi.org/project/xgboost
libsvmYou can download libsvm wheel from various web sources. One example can be found here: https://www.lfd.uci.edu/~gohlke/pythonlibs/#libsvm
coremltoolsCurrenlty coreml does not distribute coreml packaging on windows. You can install from source: pip install git+https://github.com/apple/coremltools
pysparkhttps://pypi.org/project/pyspark

有关winmltools的详细信息,可以转到Convert ML models to ONNX with WinMLTools

许可证

麻省理工学院许可证

欢迎加入QQ群-->: 979659372 Python中文网_新手群

推荐PyPI第三方库


热门话题
javascript如何从Java中的seleniun访问Chrome上的PDF查看器DOM?   如何打印字符串中的大写字母数   java使用htmlUnit处理动态html表单   java等待事件分派线程创建的GUI   java如何在屏幕上移动/动画单个子类对象而不移动其他子类?   java拟合是从PRNG到最大-最小范围的结果   java如何隐藏所有其他片段   java OpenGL ES纹理图谱失真   java排队/退队或报价/投票   java step/stp文件在另外两个点之间的距离   java如何设置水平仪栏的动画?   在Flex和Java中压缩和解压缩XML   java如何使用AutoIt安装JRE   java我想删除NavigationView布局上的一个栏   java Android Studio按钮小部件OnClick。找不到函数   java ImageView隐藏和显示   java类型的方法和构造函数未定义   swing Java UDP数据包未从客户端程序到达服务器程序   java如何在我自己的服务器上配置推送通知?