Skip to content

Sort by confidence and remove excess boxes without explicit if#10517

Merged
glenn-jocher merged 3 commits intoultralytics:masterfrom
developer0hye:patch-15
Dec 19, 2022
Merged

Sort by confidence and remove excess boxes without explicit if#10517
glenn-jocher merged 3 commits intoultralytics:masterfrom
developer0hye:patch-15

Conversation

@developer0hye
Copy link
Copy Markdown
Contributor

@developer0hye developer0hye commented Dec 16, 2022

It is related to #10502.
Signed-off-by: Yonghye Kwon developer.0hye@gmail.com

🛠️ PR Summary

Made with ❤️ by Ultralytics Actions

🌟 Summary

Optimization of Non-Max Suppression (NMS) in YOLOv5 object detection.

📊 Key Changes

  • Simplified the sorting of detection boxes based on confidence score.
  • Removed excess boxes beyond a maximum threshold (max_nms) in one line of code.

🎯 Purpose & Impact

  • Purpose: To streamline the box sorting process within NMS and improve code readability.
  • Impact: This change can potentially speed up the NMS step in object detection, leading to faster inference times. It also makes the codebase easier to understand and maintain for developers. Users can expect slightly quicker results without compromising accuracy when detecting objects in images. 🚀

@glenn-jocher glenn-jocher changed the title sort by confidence and remove excess boxes without explicit if Sort by confidence and remove excess boxes without explicit if Dec 19, 2022
@glenn-jocher glenn-jocher merged commit 5545ff3 into ultralytics:master Dec 19, 2022
@glenn-jocher
Copy link
Copy Markdown
Member

@developer0hye PR is merged. Thank you for your contributions to YOLOv5 🚀 and Vision AI ⭐

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.

2 participants