Common Models
The Nx toolkit currently offers support for image classification and object detection models. To facilitate easy deployment, we provide several guides containing instructions showing how to deploy common models such as MobileNet, ResNet, EfficientNet, ViT, LeViT for image classification, and YoloS, Yolov4, Yolov7, Yolov11 for object detection.
Key steps involved in this process include:
Incorporating necessary post-processing steps such as masking and a configurable Non-Maximum Suppression (NMS) for object detection models and Softmax for image classification models directly into the ONNX graph.
Modifying the model's input and output shapes and names as per the requirements of the AI Manager.
Making sure the exported ONNX has an Operator Set Version less than 16.
The primary objective of these guides is to equip AI developers with off-the-shelf scripts, enabling them to deploy these models within minutes. This is achieved by exporting the models to ONNX format, ensuring compatibility with the AI Manager.
Last updated