Gen_wts issue with runnung

marcoslucianops / DeepStream-Yolo

NVIDIA DeepStream SDK 7.0 / 6.4 / 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 / 5.1 implementation for YOLO models

MIT License

1.38k stars 343 forks source link

Gen_wts issue with runnung #486

Open AbdullahGM1 opened 7 months ago

AbdullahGM1 commented 7 months ago

Hello,

I have an issue with running gen_wts for YoloV8

It shows me this error:

File "gen_wts_yoloV8.py", line 5, in <module> from ultralytics.yolo.utils.torch_utils import select_device ModuleNotFoundError: No module named 'ultralytics.yolo'

How to solve this issue?

thank you

marcoslucianops commented 7 months ago

The gen_wts is outdated. Try to use the new ONNX export method.