Skip to content

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. 🚀

@glenn-jocher glenn-jocher added bug Something isn't working TODO High priority items labels Sep 25, 2022
@glenn-jocher glenn-jocher merged commit 966b0e0 into master Sep 25, 2022
@glenn-jocher glenn-jocher deleted the update/inference branch September 25, 2022 14:21
@glenn-jocher glenn-jocher removed the TODO High priority items label Sep 25, 2022
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