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

Add macos-latest runner for CoreML benchmarks #9453

Merged
merged 6 commits into from
Sep 17, 2022
Merged

Conversation

glenn-jocher
Copy link
Member

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

Signed-off-by: Glenn Jocher glenn.jocher@ultralytics.com

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhanced model output handling for segmentation in YOLOv5.

πŸ“Š Key Changes

  • Simplified the logic for handling outputs in models that include segmentation tasks.
  • The previous method of identifying the last layer's output by key has been replaced.
  • Outputs are now retrieved by reversing the order of model values which is particularly relevant for certain segmentation models.

🎯 Purpose & Impact

  • The change aims to streamline the post-processing step in YOLOv5's forward pass, especially for models that perform segmentation tasks.
  • This can potentially make the code easier to maintain and understand, as it removes the need for dynamically constructing the key based on the outputs available.
  • Users can expect a more consistent handling of outputs, which may, in turn, enhance the usability of YOLOv5 for various computer vision tasks including segmentation.
  • There might be a performance improvement in scenarios where segmentation model outputs are used, due to a more direct access to the required output tensors. πŸš€

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
@glenn-jocher glenn-jocher merged commit e8a9c5a into master Sep 17, 2022
@glenn-jocher glenn-jocher deleted the coreml_benchmarks branch September 17, 2022 22:57
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.

None yet

1 participant