OverviewHardwareSoftwareCloud
OAK Apps
DepthAI
AI
Software Stack
Troubleshooting

ON THIS PAGE

  • Overview
  • Export Format
  • CLI
  • Python API
  • NN Archive
  • CLI
  • Python API
  • Convert
  • CLI
  • Python API

Exporting

Overview

Export your trained models for downstream deployment workflows. LuxonisTrain supports direct ONNX export, NN Archive generation, and a unified conversion flow for Luxonis target platforms.

Export Format

  • ONNX: Open Neural Network Exchange format used as the main interchange and downstream conversion format.
To configure export and conversion behavior, you can specify the exporter section in the config file:
Yaml
1exporter:
2  onnx:
3    opset_version: 11
exporter.hubai and exporter.blobconverter are used by convert or ConvertOnTrainEnd, not by export alone.

CLI

Use the following command to export your model:
Command Line
1luxonis_train export --config configs/example_export.yaml --weights path/to/weights.ckpt

Python API

Python
1from luxonis_train import LuxonisModel
2
3model = LuxonisModel("configs/example_export.yaml")
4model.export(weights="path/to/weights.ckpt")

NN Archive

Create an NN Archive file for easy deployment with the DepthAI API. If you do not provide an ONNX executable, LuxonisTrain exports the model to ONNX first.The archive contains the exported model together with all the metadata needed for running the model seamlessly.

CLI

To create an NN Archive using the CLI, use the following command:
Command Line
1luxonis_train archive                         \
2  --config configs/detection_light_model.yaml \
3  --weights path/to/checkpoint.ckpt

Python API

You can also create an NN Archive using the Python API:
Python
1from luxonis_train import LuxonisModel
2
3model = LuxonisModel("configs/detection_light_model.yaml")
4model.archive(weights="path/to/checkpoint.ckpt")

Convert

convert is the unified deployment flow in LuxonisTrain. It performs:
  1. export from checkpoint to ONNX
  2. archive from ONNX to NN Archive
  3. optional platform-specific conversion for RVC2, RVC3, or RVC4
Configure this behavior through exporter.hubai or exporter.blobconverter.

CLI

Command Line
1luxonis_train convert --config configs/detection_light_model.yaml --weights path/to/checkpoint.ckpt

Python API

Python
1from luxonis_train import LuxonisModel
2
3model = LuxonisModel("configs/detection_light_model.yaml")
4archive_path, conversion_artifacts = model.convert(
5    weights="path/to/checkpoint.ckpt",
6)
convert() returns the ONNX-based NN Archive path together with a dictionary of any additional conversion artifacts, such as a platform-specific archive or legacy blob output.