This repository was archived by the owner on Jul 17, 2024. It is now read-only.

onnx_import.py fails on MobileNet reference model #228

Open

shawncal opened this issue Jul 4, 2019 · 0 comments

Contributor

shawncal commented Jul 4, 2019 •

edited

Loading

Goal: Import a MobileNet reference model from the ONNX Model Zoo.

Repro Steps:

Download the mobilenetv2-1.0.onnx model from the ONNX Model Zoo:

https://github.com/onnx/models/tree/master/models/image_classification/mobilenet

Run the onnx_import.py tool:

python E:\ELL\tools\importers\onnx\onnx_import.py mobilenetv2-1.0.onnx

Result:
ValueError: cannot reshape array of size 1000 into shape (0,newaxis)

Updates:

Same error observed with SqueezeNet:

https://github.com/onnx/models/tree/master/models/image_classification/squeezenet

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.