ARM-software / armnn

Arm NN ML Software. The code here is a read-only mirror of https://review.mlplatform.org/admin/repos/ml/armnn
https://developer.arm.com/products/processors/machine-learning/arm-nn
MIT License
1.17k stars 309 forks source link

Running YOLOv5 ONNX model with onnx parser of armnn fails with unsupported operation #768

Closed Murari007 closed 1 month ago

Murari007 commented 5 months ago

Traceback (most recent call last): File "/home/sl-rpi-5/Downloads/armnn/python/pyarmnn/examples/object_detection/yolov5_onnx_parser.py", line 114, in main(args) File "/home/sl-rpi-5/Downloads/armnn/python/pyarmnn/examples/object_detection/yolov5_onnx_parser.py", line 55, in main executor = NetworkExecutor(*exec_input_args) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/sl-rpi-5/Downloads/armnn/python/pyarmnn/examples/object_detection/network_executor.py", line 22, in init self.network_id, self.runtime, self.input_binding_info, self.output_binding_info = self.create_network() ^^^^^^^^^^^^^^^^^^^^^ File "/home/sl-rpi-5/Downloads/armnn/python/pyarmnn/examples/object_detection/network_executor.py", line 60, in create_network network = parser.CreateNetworkFromBinaryFile(self.model_file) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/sl-rpi-5/armnn-pi/myenv/lib/python3.11/site-packages/pyarmnn/_generated/pyarmnn_onnxparser.py", line 656, in CreateNetworkFromBinaryFile return _pyarmnn_onnxparser.IOnnxParser_CreateNetworkFromBinaryFile(self, graphFile) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ RuntimeError: Unsupported operation Mul for node '/model.0/act/Mul' at function LoadGraph [/home/sl-rpi-5/armnn-pi/armnn/build-tool/scripts/source/armnn/src/armnnOnnxParser/OnnxParser.cpp:966]

catcor01 commented 5 months ago

Hello,

Your issue is somewhat similar to the following one: https://github.com/ARM-software/armnn/issues/761.

Mul support has not been added to the ONNX parser. The work to add this support is on our radar but has not been prioritized in the near future. I can make 3 suggestions to get things running on your side:

Kind Regards, Cathal.

Murari007 commented 5 months ago

Hello @catcor01

Thanks for the reply. I will try the suggested alternative methods & get back soon.

Kind Regads, Murari