Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

onnx convert tensorrt with dynamic issue #13360

Open
1 of 2 tasks
fenghuoxiguozu opened this issue Oct 15, 2024 · 1 comment
Open
1 of 2 tasks

onnx convert tensorrt with dynamic issue #13360

fenghuoxiguozu opened this issue Oct 15, 2024 · 1 comment
Labels
bug Something isn't working

Comments

@fenghuoxiguozu
Copy link

Search before asking

  • I have searched the YOLOv5 issues and found no similar bug report.

YOLOv5 Component

No response

Bug

find same issue https://github.com/ultralytics/yolov5/issues/9764

image

Environment

yolov5-7.0 torch1.13+ cu116 + tensorrt8.24

Minimal Reproducible Example

onnx export success:
python export.py --weights yolov5m.pt --include onnx --dynamic --simplify --opset 12

trt convert error:
trtexec --onnx=yolov5m.onnx --minShapes=images:1x3x640x640 --optShapes=images:10x3x640x640 --maxShapes=images:10x3x640x640 --saveEngine=yolov5m.engine --fp16

Additional

No response

Are you willing to submit a PR?

  • Yes I'd like to help by submitting a PR!
@fenghuoxiguozu fenghuoxiguozu added the bug Something isn't working label Oct 15, 2024
@pderrenger
Copy link
Member

@fenghuoxiguozu please ensure you're using the latest YOLOv5 version, as updates may resolve your issue. For guidance on TensorRT conversion, refer to our documentation at https://docs.ultralytics.com/integrations/tensorrt/. If the problem persists, consider checking the TensorRT logs for more details.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

No branches or pull requests

2 participants