Import CustomVision.ai ONNX Models - Githubissues

microsoft / ELL

Embedded Learning Library

https://microsoft.github.io/ELL

Other

2.29k stars 295 forks source link

Import CustomVision.ai ONNX Models #227

Open shawncal opened 5 years ago

shawncal commented 5 years ago

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