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

Single-line TRT dynamic assertion #8871

Merged
merged 1 commit into from
Aug 4, 2022
Merged

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Aug 4, 2022

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhanced error messaging for model input size mismatches in YOLOv5.

πŸ“Š Key Changes

  • Simplified the error message when input image shape doesn't match the expected model input shape.
  • Adjusted the condition to accommodate dynamic input shapes.

🎯 Purpose & Impact

  • 🎨 Easier to understand: The error message is now clearer for users encountering size mismatch issues.
  • ✨ Dynamic shape support: The new assert condition properly handles models with dynamic input sizes, improving user experience for a broader range of model configurations.

@glenn-jocher glenn-jocher self-assigned this Aug 4, 2022
@glenn-jocher glenn-jocher merged commit 731a2f8 into master Aug 4, 2022
@glenn-jocher glenn-jocher deleted the glenn-jocher-patch-1 branch August 4, 2022 21:34
ctjanuhowski pushed a commit to ctjanuhowski/yolov5 that referenced this pull request Sep 8, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant