DeepStream-Yolo icon indicating copy to clipboard operation
DeepStream-Yolo copied to clipboard

Gen_wts issue with runnung

Open AbdullahGM1 opened this issue 1 year ago • 1 comments

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

AbdullahGM1 avatar Nov 23 '23 07:11 AbdullahGM1

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

marcoslucianops avatar Nov 24 '23 00:11 marcoslucianops