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.39k stars 344 forks source link

Failed to generate engine file #434

Open ghty11468 opened 10 months ago

ghty11468 commented 10 months ago

8e95ee19b1997032ce8cc71e57b5f70

I think there is no problem with the generation process of the onnx file, and I did not use the -- dynamic command. The command I used to generate onnx files is: python3 export_ yoloV5. py - w yolov5s. pt - s 320. However, this error occurred during the generation of the engine file.

marcoslucianops commented 10 months ago

Try with ONNX model exported with command below:

python3 export_yoloV5.py -w yolov5s.pt -s 320 --opset 11
ghty11468 commented 10 months ago

尝试使用以下命令导出的 ONNX 模型:

python3 export_yoloV5.py -w yolov5s.pt -s 320 --opset 11

I have already tried. The same error will still occur

marcoslucianops commented 10 months ago

Sorry I saw the log now, probably you don't need the --opset; The log is saying that the exported ONNX model is with dynamic range. You should not get this error if you are using the command without the --dynamic arg.

ghty11468 commented 10 months ago

Sorry I saw the log now, probably you don't need the ; The log is saying that the exported ONNX model is with dynamic range. You should not get this error if you are using the command without the arg.--opset``--dynamic

I did not use “-- dynamic”, but it still shows this error and I don't know how to solve it

marcoslucianops commented 10 months ago

What's your DeepStream version?

ghty11468 commented 10 months ago

What's your DeepStream version?

6.0

marcoslucianops commented 10 months ago

I just tested the DeepStream 6.0 and I only get this error when I use ONNX model exported with --dynamic. Please check again your files.

ghty11468 commented 10 months ago

我刚刚测试了 DeepStream 6.0,只有当我使用使用 . 导出的 ONNX 模型时,我才会收到此错误。请再次检查您的文件。--dynamic

I am very certain that I did not use '-- dynamic', and I am also puzzled why this error is

llzcyy commented 2 months ago

i have the same problem,deepstream is 5.1