Hi,
I’m trying to convert an onnx file that was exported from GitHub - baudm/parseq: Scene Text Recognition with Permuted Autoregressive Sequence Models (ECCV 2022) project and then simplified, but I get this error:
TypeError: ‘NoneType’ object is not subscriptable
This is a part of the stack trace:
File "/home/hailo/.local/lib/python3.10/site-packages/hailo_sdk_client/model_translator/onnx_translator/onnx_translator.py", line 385, in _layer_callback_from_vertex
if vertex.is_null_operation() and not is_flattened_global_maxpool:
File "/home/hailo/.local/lib/python3.10/site-packages/hailo_sdk_client/model_translator/onnx_translator/onnx_graph.py", line 5307, in is_null_operation
or (self.op == "ReduceMean" and self.is_null_reduce_mean())
File "/home/hailo/.local/lib/python3.10/site-packages/hailo_sdk_client/model_translator/onnx_translator/onnx_graph.py", line 5342, in is_null_reduce_mean
axes = self._convert_axes_to_nhwc(axes_info)
File "/home/hailo/.local/lib/python3.10/site-packages/hailo_sdk_client/model_translator/onnx_translator/onnx_graph.py", line 2250, in _convert_axes_to_nhwc
return [nchw_to_nhwc_axis_mapping[self.input_format[axis]] for axis in axes]
File "/home/hailo/.local/lib/python3.10/site-packages/hailo_sdk_client/model_translator/onnx_translator/onnx_graph.py", line 2250, in <listcomp>
return [nchw_to_nhwc_axis_mapping[self.input_format[axis]] for axis in axes]
TypeError: 'NoneType' object is not subscriptable
My code
from hailo_sdk_client import ClientRunner
chosen_hw_arch = "hailo8l"
onnx_model_name = "parseq_tiny_fp32_simple"
onnx_path = f"/onnx/{onnx_model_name}.onnx"
runner = ClientRunner(hw_arch=chosen_hw_arch)
hn, npz = runner.translate_onnx_model(
onnx_path,
onnx_model_name,
start_node_names=["images"],
end_node_names=["5293"],
disable_shape_inference=True,
)
hailo_model_har_name = f"/onnx/{onnx_model_name}_hailo_model.har"
runner.save_har(hailo_model_har_name)
I have traced the error back to this node:
I am running it inside a docker container with these packages:
- python3.10
- hailo_dataflow_compiler-3.31.0-py3-none-linux_x86_64.whl
- hailort-4.21.0-cp310-cp310-linux_x86_64.whl
- hailort_4.21.0_amd64.deb
Any suggestions on how to solve this?
Thank you in advance.