search

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.4k stars 345 forks source link

TypeError: attempt_load() got an unexpected keyword argument 'device' #360

Open trameshape opened 1 year ago

trameshape commented 1 year ago

When running export_yoloV5.py, It supported an error, " TypeError: attempt_load() got an unexpected keyword argument 'device' "

marcoslucianops commented 1 year ago

Please use the master branch of the ultralytics/yolov5 to convert the model (even using older versions).

ghty11468 commented 11 months ago

Please use the branch of the to convert the model (even using older versions).master``ultralytics/yolov5

Is it necessary to use the model obtained from the old version of the YOLOv5 project? Isn't yolov5 6.0 okay?

ghty11468 commented 11 months ago

Please use the branch of the to convert the model (even using older versions).master``ultralytics/yolov5

I am also reporting errors like this. How can I solve it?

marcoslucianops commented 10 months ago

Any version of the YOLOv5 model will work in the export to ONNX. I just tested and it's working normally.

shuzhenglin commented 8 months ago

@ghty11468 sir,have you solve this problem.I also get this error when i use export_yoloV5.So use the export.py in yolov5 6.0.The generated ONNX file runs out of DeepStream without a Bounding Box in the container