Hailo Model Zoo in google colab

Hello,

I have trained an instance segmentation model using yolov8n and have an working ONNX file. I manage to install the DFC 3.30 on google colab along with the Model Zoo.

When I run the following command
my_env/bin/hailomz info mobilenet_v1

I can see that hailmz is working. So I proceed to use the compile instruction for my onnx
This is the full command I execute

!my_env/bin/hailomz compile --ckpt /content/best.onnx --calib-path /content/valid/images --yaml /content/hailo_model_zoo/hailo_model_zoo/cfg/networks/yolov8n_seg.yaml

However, I get the following error

Traceback (most recent call last):
File “/content/my_env/bin/hailomz”, line 33, in
sys.exit(load_entry_point(‘hailo-model-zoo’, ‘console_scripts’, ‘hailomz’)())
File “/content/hailo_model_zoo/hailo_model_zoo/main.py”, line 122, in main
run(args)
File “/content/hailo_model_zoo/hailo_model_zoo/main.py”, line 101, in run
from hailo_model_zoo.main_driver import compile, evaluate, optimize, parse, profile
File “/content/hailo_model_zoo/hailo_model_zoo/main_driver.py”, line 16, in
from hailo_model_zoo.core.main_utils import (
File “/content/hailo_model_zoo/hailo_model_zoo/core/main_utils.py”, line 18, in
from hailo_model_zoo.core.infer import infer_factory
File “/content/hailo_model_zoo/hailo_model_zoo/core/infer/infer_factory.py”, line 14, in
discovered_plugins = {name: importlib.import_module(name) for _, name, _ in iter_namespace(hailo_model_zoo.core.infer)}
File “/content/hailo_model_zoo/hailo_model_zoo/core/infer/infer_factory.py”, line 14, in
discovered_plugins = {name: importlib.import_module(name) for _, name, _ in iter_namespace(hailo_model_zoo.core.infer)}
File “/usr/lib/python3.10/importlib/**init**.py”, line 126, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
File “/content/hailo_model_zoo/hailo_model_zoo/core/infer/hw_infer_utils.py”, line 6, in
from hailo_platform import (
ModuleNotFoundError: No module named ‘hailo_platform’

I also tried to install HailoRT but that breaks hailomz and hailoDFC so I removed it from my virtual environment.

Hi @aldiaz,

It seems the issue might be due to missing pyHailoRT. Could you try installing just the Python package first.
https://hailo.ai/developer-zone/documentation/hailort-v4-20-0/?sp_referrer=install/install.html#ubuntu-installer-requirements

Let me know if this resolves your issue!

Hello I managed, to convert my weights to the hef file. However, something that I see is that the performance is really bad compared to the ONNX weights. It’s not detecting anything I trained my model with around 3000 images. How can I improve the performance of the hef weights?