What is ONNX

2022年6月06日
AI

ONNX is a machine learning framework which acts like a medium to convert between different machine learning frameworks.

ONNX is designed to enable framework interoperability. There are many great machine learning libraries in multiple languages — PyTorch, TensorFlow, MXNet, and Caffe are just a few of the most popular ones in recent years, but there are many others.
The idea is that you can use one stack of tools to train your model and use another for inference and prediction to expand your model.

For example, once you’ve trained your model, you need to deploy it to a new iOS app so that anyone with a previously trained model can see the safety of their food. I first trained my model with PyTorch, but iOS expects to use CoreML for use within the app. ONNX is an intermediate representation of the model that allows you to easily move from one environment to another.

With tools such as ONNX-CoreML, you can now easily convert pre-trained models to files, import them into XCode, and integrate them seamlessly with your application.

This is just an overview of what ONNX is about. I will talk more detail in future.

Yuuma