Pre-Trained ML Models

1 Introduction

The ONNX community provides an ML model repository, called ONNX Model Zoo, where common computer vision and language models can be found. The ONNX Model Zoo is a collection of pre-trained, state-of-the-art models in the ONNX format contributed by community members. Accompanying each model are Jupyter notebooks for model training and running inference with the trained model. The notebooks are written in Python and include links to the training dataset, as well as references to the original paper that describes the model architecture. All the ONNX models in the ONNX Zoo should be compatible with the Mendix ML Kit. You can pick up any model from this repository, customize it with your own data and integrate it in your Mendix app using the ML Kit.

2 ONNX Model Zoo

Below, you can find the list of ONNX models in the ONNX Model Zoo. You can use these pre-trained models to build your business use cases similar to the ones listed above, or other types of use cases for your business.

Vision

• Image Classification
• Object Detection and Image Segmentation
• Body, Face, and Gesture Analysis
• Image Manipulation

Language

• Machine Comprehension
• Machine Translation
• Language Modelling

Other

• Visual Question Answering and Dialogue
• Speech and Audio Processing
• Other interesting models

2.1 Additional Pre-Trained Models

Hugging Face provides open source pre-trained models (including datasets). You can find hundreds of ONNX models to perform tasks on different modalities such as text, vision, and audio.

3 Integrate Pre-Built Models

To integrate a pre-built (propriety or public) ML model into a Mendix app, consider the following:

• Compatibility of the model internal representation (IR) version with the ML Kit
• Compatibility of the model OpSets with the ML Kit

Then, follow the instructions to import an ML model and create the ML mapping document in the Using ML Kit document.For details on versioning schemes, see the Versioning section of Using ML Kit.