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

Detect() and Segment() fixes for CoreML and Paddle #9458

Merged
merged 2 commits into from
Sep 17, 2022

Conversation

glenn-jocher
Copy link
Member

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

Eliminates .clone() that was causing Paddle export issues and inplace .sigmoid_() that was causing CoreML export issues.

Signed-off-by: Glenn Jocher [email protected]

🛠️ PR Summary

Made with ❤️ by Ultralytics Actions

🌟 Summary

Enhancements in YOLOv5's output processing for object detection and segmentation.

📊 Key Changes

  • Improved handling of bounding box (xy and wh) predictions and class confidence (conf) computations.
  • Added support for a new segment type in the output processing, which now manages boxes and masks together.
  • Code refactoring to cleanly separate the processing logic for Detect (boxes only) and Segment (boxes + masks) output types.

🎯 Purpose & Impact

  • 🎨 The update provides a clearer separation of detection and segmentation logic, making the codebase easier to understand and maintain.
  • 📈 It prepares YOLOv5 for better segmentation capabilities, potentially increasing its use in applications that require both object detection and segmentation, such as advanced computer vision tasks in autonomous driving or medical image analysis.
  • 🚀 For users, this means improved functionality and potentially more accurate and versatile models that can handle complex tasks involving not just identifying objects, but also delineating their exact shape and size within an image.

@glenn-jocher glenn-jocher changed the title Detect() and Segment() fixes for CoreML and Paddle Detect() and Segment() fixes for CoreML and Paddle Sep 17, 2022
@glenn-jocher glenn-jocher self-assigned this Sep 17, 2022
@glenn-jocher glenn-jocher merged commit 0608374 into master Sep 17, 2022
@glenn-jocher glenn-jocher deleted the glenn-jocher-patch-1 branch September 17, 2022 21:18
@glenn-jocher
Copy link
Member Author

Resolves #9439

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.

Paddle export bug: torch tensor .clone() is not supported
1 participant