2019-01-11 19:08:33 +00:00
.. note::
:class: sphx-glr-download-link-note
Click :ref:`here <sphx_glr_download_auto_examples_plot_common_errors.py>` to download the full example code
.. rst-class:: sphx-glr-example-title
.. _sphx_glr_auto_examples_plot_common_errors.py:
.. _l-example-common-error:
Common errors with onnxruntime
==============================
This example looks into several common situations
in which *onnxruntime* does not return the model
prediction but raises an exception instead.
It starts by loading the model trained in example
:ref:`l-logreg-example` which produced a logistic regression
trained on *Iris* datasets. The model takes
a vector of dimension 2 and returns a class among three.
2019-12-20 21:35:58 +00:00
.. code-block:: python
2019-01-11 19:08:33 +00:00
import onnxruntime as rt
2019-10-30 16:35:16 +00:00
from onnxruntime.capi.onnxruntime_pybind11_state import InvalidArgument
2019-01-11 19:08:33 +00:00
import numpy
from onnxruntime.datasets import get_example
example2 = get_example("logreg_iris.onnx")
sess = rt.InferenceSession(example2)
input_name = sess.get_inputs()[0].name
output_name = sess.get_outputs()[0].name
The first example fails due to *bad types*.
*onnxruntime* only expects single floats (4 bytes)
and cannot handle any other kind of floats.
2019-12-20 21:35:58 +00:00
.. code-block:: python
2019-01-11 19:08:33 +00:00
try:
x = numpy.array([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]], dtype=numpy.float64)
sess.run([output_name], {input_name: x})
except Exception as e:
print("Unexpected type")
print("{0}: {1}".format(type(e), e))
.. rst-class:: sphx-glr-script-out
Out:
.. code-block:: none
Unexpected type
2019-12-20 21:35:58 +00:00
<class 'onnxruntime.capi.onnxruntime_pybind11_state.InvalidArgument'>: [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Unexpected input data type. Actual: (class onnxruntime::PrimitiveDataType<double>) , expected: (class onnxruntime::PrimitiveDataType<float>)
2019-01-11 19:08:33 +00:00
2019-08-02 01:12:59 +00:00
2019-01-11 19:08:33 +00:00
The model fails to return an output if the name
is misspelled.
2019-12-20 21:35:58 +00:00
.. code-block:: python
2019-01-11 19:08:33 +00:00
try:
x = numpy.array([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]], dtype=numpy.float32)
sess.run(["misspelled"], {input_name: x})
except Exception as e:
print("Misspelled output name")
print("{0}: {1}".format(type(e), e))
.. rst-class:: sphx-glr-script-out
Out:
.. code-block:: none
Misspelled output name
2019-12-20 21:35:58 +00:00
<class 'onnxruntime.capi.onnxruntime_pybind11_state.InvalidArgument'>: [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Invalid Output Name:misspelled
2019-01-11 19:08:33 +00:00
The output name is optional, it can be replaced by *None*
and *onnxruntime* will then return all the outputs.
2019-12-20 21:35:58 +00:00
.. code-block:: python
2019-01-11 19:08:33 +00:00
x = numpy.array([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]], dtype=numpy.float32)
2019-10-30 16:35:16 +00:00
try:
res = sess.run(None, {input_name: x})
print("All outputs")
print(res)
except (RuntimeError, InvalidArgument) as e:
print(e)
2019-01-11 19:08:33 +00:00
.. rst-class:: sphx-glr-script-out
Out:
.. code-block:: none
2019-12-20 21:35:58 +00:00
All outputs
[array([0, 0, 0], dtype=int64), [{0: 0.950599730014801, 1: 0.027834169566631317, 2: 0.02156602405011654}, {0: 0.9974970817565918, 1: 5.6299926654901356e-05, 2: 0.0024466661270707846}, {0: 0.9997311234474182, 1: 1.1918064757310276e-07, 2: 0.00026869276189245284}]]
2019-01-11 19:08:33 +00:00
2019-08-02 01:12:59 +00:00
The same goes if the input name is misspelled.
2019-01-11 19:08:33 +00:00
2019-12-20 21:35:58 +00:00
.. code-block:: python
2019-01-11 19:08:33 +00:00
try:
x = numpy.array([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]], dtype=numpy.float32)
sess.run([output_name], {"misspelled": x})
except Exception as e:
print("Misspelled input name")
print("{0}: {1}".format(type(e), e))
.. rst-class:: sphx-glr-script-out
Out:
.. code-block:: none
Misspelled input name
2019-10-30 16:35:16 +00:00
<class 'onnxruntime.capi.onnxruntime_pybind11_state.InvalidArgument'>: [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Invalid Feed Input Name:misspelled
2019-01-11 19:08:33 +00:00
*onnxruntime* does not necessarily fail if the input
dimension is a multiple of the expected input dimension.
2019-12-20 21:35:58 +00:00
.. code-block:: python
2019-01-11 19:08:33 +00:00
for x in [
numpy.array([1.0, 2.0, 3.0, 4.0], dtype=numpy.float32),
numpy.array([[1.0, 2.0, 3.0, 4.0]], dtype=numpy.float32),
numpy.array([[1.0, 2.0], [3.0, 4.0]], dtype=numpy.float32),
numpy.array([1.0, 2.0, 3.0], dtype=numpy.float32),
numpy.array([[1.0, 2.0, 3.0]], dtype=numpy.float32),
]:
2019-10-30 16:35:16 +00:00
try:
r = sess.run([output_name], {input_name: x})
print("Shape={0} and predicted labels={1}".format(x.shape, r))
except (RuntimeError, InvalidArgument) as e:
print("ERROR with Shape={0} - {1}".format(x.shape, e))
2019-01-11 19:08:33 +00:00
for x in [
numpy.array([1.0, 2.0, 3.0, 4.0], dtype=numpy.float32),
numpy.array([[1.0, 2.0, 3.0, 4.0]], dtype=numpy.float32),
numpy.array([[1.0, 2.0], [3.0, 4.0]], dtype=numpy.float32),
numpy.array([1.0, 2.0, 3.0], dtype=numpy.float32),
numpy.array([[1.0, 2.0, 3.0]], dtype=numpy.float32),
]:
2019-10-30 16:35:16 +00:00
try:
r = sess.run(None, {input_name: x})
print("Shape={0} and predicted probabilities={1}".format(x.shape, r[1]))
except (RuntimeError, InvalidArgument) as e:
print("ERROR with Shape={0} - {1}".format(x.shape, e))
2019-01-11 19:08:33 +00:00
.. rst-class:: sphx-glr-script-out
Out:
.. code-block:: none
2019-10-30 16:35:16 +00:00
ERROR with Shape=(4,) - [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Invalid rank for input: float_input Got: 1 Expected: 2 Please fix either the inputs or the model.
2019-12-20 21:35:58 +00:00
ERROR with Shape=(1, 4) - [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Got invalid dimensions for input: float_input for the following indices
index: 0 Got: 1 Expected: 3
index: 1 Got: 4 Expected: 2
Please fix either the inputs or the model.
2019-10-30 16:35:16 +00:00
ERROR with Shape=(2, 2) - [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Got invalid dimensions for input: float_input for the following indices
2019-12-20 21:35:58 +00:00
index: 0 Got: 2 Expected: 3
2019-10-30 16:35:16 +00:00
Please fix either the inputs or the model.
ERROR with Shape=(3,) - [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Invalid rank for input: float_input Got: 1 Expected: 2 Please fix either the inputs or the model.
ERROR with Shape=(1, 3) - [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Got invalid dimensions for input: float_input for the following indices
2019-12-20 21:35:58 +00:00
index: 0 Got: 1 Expected: 3
index: 1 Got: 3 Expected: 2
2019-10-30 16:35:16 +00:00
Please fix either the inputs or the model.
ERROR with Shape=(4,) - [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Invalid rank for input: float_input Got: 1 Expected: 2 Please fix either the inputs or the model.
2019-12-20 21:35:58 +00:00
ERROR with Shape=(1, 4) - [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Got invalid dimensions for input: float_input for the following indices
index: 0 Got: 1 Expected: 3
index: 1 Got: 4 Expected: 2
Please fix either the inputs or the model.
2019-10-30 16:35:16 +00:00
ERROR with Shape=(2, 2) - [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Got invalid dimensions for input: float_input for the following indices
2019-12-20 21:35:58 +00:00
index: 0 Got: 2 Expected: 3
2019-10-30 16:35:16 +00:00
Please fix either the inputs or the model.
ERROR with Shape=(3,) - [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Invalid rank for input: float_input Got: 1 Expected: 2 Please fix either the inputs or the model.
ERROR with Shape=(1, 3) - [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Got invalid dimensions for input: float_input for the following indices
2019-12-20 21:35:58 +00:00
index: 0 Got: 1 Expected: 3
index: 1 Got: 3 Expected: 2
2019-10-30 16:35:16 +00:00
Please fix either the inputs or the model.
2019-01-11 19:08:33 +00:00
2019-08-02 01:12:59 +00:00
2019-01-11 19:08:33 +00:00
It does not fail either if the number of dimension
is higher than expects but produces a warning.
2019-12-20 21:35:58 +00:00
.. code-block:: python
2019-01-11 19:08:33 +00:00
for x in [
numpy.array([[[1.0, 2.0], [3.0, 4.0]]], dtype=numpy.float32),
numpy.array([[[1.0, 2.0, 3.0]]], dtype=numpy.float32),
numpy.array([[[1.0, 2.0]], [[3.0, 4.0]]], dtype=numpy.float32),
]:
2019-10-30 16:35:16 +00:00
try:
r = sess.run([output_name], {input_name: x})
print("Shape={0} and predicted labels={1}".format(x.shape, r))
except (RuntimeError, InvalidArgument) as e:
print("ERROR with Shape={0} - {1}".format(x.shape, e))
2019-01-11 19:08:33 +00:00
.. rst-class:: sphx-glr-script-out
Out:
.. code-block:: none
2019-10-30 16:35:16 +00:00
ERROR with Shape=(1, 2, 2) - [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Invalid rank for input: float_input Got: 3 Expected: 2 Please fix either the inputs or the model.
ERROR with Shape=(1, 1, 3) - [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Invalid rank for input: float_input Got: 3 Expected: 2 Please fix either the inputs or the model.
ERROR with Shape=(2, 1, 2) - [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Invalid rank for input: float_input Got: 3 Expected: 2 Please fix either the inputs or the model.
2019-01-11 19:08:33 +00:00
2019-08-02 01:12:59 +00:00
2019-12-20 21:35:58 +00:00
**Total running time of the script:** ( 0 minutes 0.049 seconds)
2019-01-11 19:08:33 +00:00
.. _sphx_glr_download_auto_examples_plot_common_errors.py:
.. only :: html
.. container:: sphx-glr-footer
:class: sphx-glr-footer-example
.. container:: sphx-glr-download
:download:`Download Python source code: plot_common_errors.py <plot_common_errors.py>`
.. container:: sphx-glr-download
:download:`Download Jupyter notebook: plot_common_errors.ipynb <plot_common_errors.ipynb>`
.. only:: html
.. rst-class:: sphx-glr-signature
2019-10-30 16:35:16 +00:00
`Gallery generated by Sphinx-Gallery <https://sphinx-gallery.readthedocs.io>`_