Skip to main content

Microsoft ML Tool to convert machine learning models to ONNX for use with Windows Machine Learning

Project description

Introduction

WinMLTools enables you to convert models from different machine learning toolkits into ONNX for use with Windows Machine Learning. Currently the following toolkits are supported:

  • Apple CoreML

  • scikit-learn (subset of models convertible to ONNX)

  • LibSVM

  • XGBoost

Install

pip install winmltools

Dependancies

This converter package extends the functionalities of ONNXMLTools.

scikit-learn is needed to convert a scikit-learn model, coremltools for Apple CoreML.

Example

Here is a simple example to convert a CoreML model:

import winmltools
import coremltools

model_coreml = coremltools.utils.load_spec("image_recognition.mlmodel")
model_onnx = winmltools.convert.convert_coreml(model_coreml, "Image_Reco")

# Save as text
winmltools.utils.save_text(model_onnx, "image_recognition.json")

# Save as protobuf
winmltools.utils.save_model(model_onnx, "image_recognition.onnx")

License

MIT License

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Built Distribution

winmltools-0.1.0.5069-py2.py3-none-any.whl (20.1 kB view hashes)

Uploaded py2 py3

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor NVIDIA NVIDIA PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page