I trained my own model using the Ultralytics yolo8n-obb. I successfully converted the .pt file to .onnx. However, when I try to convert the model to HEF using this code “hailomz compile --ckpt=newyolo8.onnx --hw-arch hailo8l --calib-path /home/reaganvernengo/hailodfc/train/clean_images --yaml /home/reaganvernengo/hailo_model_zoo/hailo_model_zoo/cfg/networks/newyolov8n.yaml --classes 2 --performance” I get this error:
Start run for network yolov8n …
Initializing the hailo8l runner…
[info] Translation started on ONNX model yolov8n
[info] Restored ONNX model yolov8n (completion time: 00:00:00.06)
[info] Extracted ONNXRuntime meta-data for Hailo model (completion time: 00:00:00.22)
[info] Simplified ONNX model for a parsing retry attempt (completion time: 00:00:00.51)
Traceback (most recent call last):
File “/home/reaganvernengo/hailodfc/lib/python3.10/site-packages/hailo_sdk_client/sdk_backend/parser/parser.py”, line 220, in translate_onnx_model
parsing_results = self._parse_onnx_model_to_hn(
File “/home/reaganvernengo/hailodfc/lib/python3.10/site-packages/hailo_sdk_client/sdk_backend/parser/parser.py”, line 300, in _parse_onnx_model_to_hn
return self.parse_model_to_hn(
File “/home/reaganvernengo/hailodfc/lib/python3.10/site-packages/hailo_sdk_client/sdk_backend/parser/parser.py”, line 340, in parse_model_to_hn
converter = ONNXConverter(
File “/home/reaganvernengo/hailodfc/lib/python3.10/site-packages/hailo_sdk_client/model_translator/onnx_translator/onnx_translator.py”, line 170, in init
super().init(
File “/home/reaganvernengo/hailodfc/lib/python3.10/site-packages/hailo_sdk_client/model_translator/edge_nn_translator.py”, line 27, in init
super().init(graph, start_node_names, end_node_names)
File “/home/reaganvernengo/hailodfc/lib/python3.10/site-packages/hailo_sdk_client/model_translator/translator.py”, line 51, in init
self._calculate_valid_subgraph_scope()
File “/home/reaganvernengo/hailodfc/lib/python3.10/site-packages/hailo_sdk_client/model_translator/translator.py”, line 388, in _calculate_valid_subgraph_scope
current_vertex.in_valid_subgraph = True
AttributeError: ‘NoneType’ object has no attribute ‘in_valid_subgraph’
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File “/home/reaganvernengo/hailodfc/bin/hailomz”, line 33, in
sys.exit(load_entry_point(‘hailo-model-zoo’, ‘console_scripts’, ‘hailomz’)())
File “/home/reaganvernengo/hailo_model_zoo/hailo_model_zoo/main.py”, line 122, in main
run(args)
File “/home/reaganvernengo/hailo_model_zoo/hailo_model_zoo/main.py”, line 111, in run
return handlersargs.command
File “/home/reaganvernengo/hailo_model_zoo/hailo_model_zoo/main_driver.py”, line 250, in compile
_ensure_optimized(runner, logger, args, network_info)
File “/home/reaganvernengo/hailo_model_zoo/hailo_model_zoo/main_driver.py”, line 73, in _ensure_optimized
_ensure_parsed(runner, logger, network_info, args)
File “/home/reaganvernengo/hailo_model_zoo/hailo_model_zoo/main_driver.py”, line 108, in _ensure_parsed
parse_model(runner, network_info, ckpt_path=args.ckpt_path, results_dir=args.results_dir, logger=logger)
File “/home/reaganvernengo/hailo_model_zoo/hailo_model_zoo/core/main_utils.py”, line 97, in parse_model
model_name = translate_model(runner, network_info, ckpt_path, tensor_shapes=start_node_shapes)
File “/home/reaganvernengo/hailo_model_zoo/hailo_model_zoo/utils/parse_utils.py”, line 28, in translate_model
runner.translate_onnx_model(
File “/home/reaganvernengo/hailodfc/lib/python3.10/site-packages/hailo_sdk_common/states/states.py”, line 16, in wrapped_func
return func(self, *args, **kwargs)
File “/home/reaganvernengo/hailodfc/lib/python3.10/site-packages/hailo_sdk_client/runner/client_runner.py”, line 1158, in translate_onnx_model
parser.translate_onnx_model(
File “/home/reaganvernengo/hailodfc/lib/python3.10/site-packages/hailo_sdk_client/sdk_backend/parser/parser.py”, line 260, in translate_onnx_model
parsing_results = self._parse_onnx_model_to_hn(
File “/home/reaganvernengo/hailodfc/lib/python3.10/site-packages/hailo_sdk_client/sdk_backend/parser/parser.py”, line 300, in _parse_onnx_model_to_hn
return self.parse_model_to_hn(
File “/home/reaganvernengo/hailodfc/lib/python3.10/site-packages/hailo_sdk_client/sdk_backend/parser/parser.py”, line 340, in parse_model_to_hn
converter = ONNXConverter(
File “/home/reaganvernengo/hailodfc/lib/python3.10/site-packages/hailo_sdk_client/model_translator/onnx_translator/onnx_translator.py”, line 170, in init
super().init(
File “/home/reaganvernengo/hailodfc/lib/python3.10/site-packages/hailo_sdk_client/model_translator/edge_nn_translator.py”, line 27, in init
super().init(graph, start_node_names, end_node_names)
File “/home/reaganvernengo/hailodfc/lib/python3.10/site-packages/hailo_sdk_client/model_translator/translator.py”, line 51, in init
self._calculate_valid_subgraph_scope()
File “/home/reaganvernengo/hailodfc/lib/python3.10/site-packages/hailo_sdk_client/model_translator/translator.py”, line 388, in _calculate_valid_subgraph_scope
current_vertex.in_valid_subgraph = True
AttributeError: ‘NoneType’ object has no attribute ‘in_valid_subgraph’
Here is my .YAML file
and my JSON file:
And the end of my Netron figure:
Any suggestions would be amazing!