Import CustomVision.ai ONNX Models

[shawncal]

Goal: import an ONNX model exported from CustomVision.ai

Repro Steps
*NOTE: Pre-trained ONNX model attached, if you want to skip training your own

Train a new model:

• Go to https://www.customvision.ai
• Create an image classifier (mobile) model
• Train the model
• From the "Perfomance" tab, click the download button captioned "export selected iteration"
• Choose "ONNX" and version 1.2, download the model to a local folder

Import ONNX model

• Run ELL's onnx_import.py script:

  python E:\ELL\tools\importers\onnx\onnx_import.py cv_model.onnx

Result:
Exception: Unimplemented node type: ImageScaler

cv_model.zip