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

TensorRT detect.py inference fix #9581

Merged
merged 5 commits into from
Sep 25, 2022
Merged

TensorRT detect.py inference fix #9581

merged 5 commits into from
Sep 25, 2022

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Sep 25, 2022

May resolve #9578

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhanced testing procedures and visualization improvements for various models in Ultralytics' YOLOv5 repository.

πŸ“Š Key Changes

  • Extended Continuous Integration Tests: Added tests for exporting .onnx models and subsequent predictions for classification, detection, and segmentation tasks.
  • Batch Size Standardization: Unified batch size initialization to 1 for all tasks and set dynamic batch sizes when using webcam data sources.
  • Visualization Adjustment: Commented out the drawing of segment polygons during segmentation prediction to possibly remove or update visualization code.

🎯 Purpose & Impact

  • Improved Reliability: Testing models in .onnx format ensures compatibility, enhancing trust in model exports for deployment.
  • Flexible Input Sources: Adjusting batch sizes dynamically caters to variable input sources like webcams, allowing for more robust and versatile usage.
  • Streamlined Visualization: Temporarily disabling polygon drawing could indicate a move to simplify or refactor visualization for better clarity or performance.

These changes can lead to more stable software and a better experience for developers and end-users working with different model formats and input sources. πŸš€

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
@glenn-jocher glenn-jocher added bug Something isn't working TODO labels Sep 25, 2022
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
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

Successfully merging this pull request may close these issues.

Tensorrt model inference not working on folder of images and videos
1 participant