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

box candidates thresholds in hyp #2869

Closed
wants to merge 1 commit into from
Closed

Conversation

ml5ah
Copy link
Contributor

@ml5ah ml5ah commented Apr 20, 2021

@glenn-jocher
In response to issue #2521

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhanced augmentation filtering with new hyperparameters for better training stability.

πŸ“Š Key Changes

  • Added three new hyperparameters to the hyp.finetune.yaml and hyp.scratch.yaml configuration files: ar_thr, area_thr, and wh_thr.
  • Integrated these new hyperparameters into dataset augmentation functions within datasets.py.

🎯 Purpose & Impact

  • 🎯 The addition of ar_thr (aspect ratio threshold), area_thr (area threshold), and wh_thr (width/height threshold) helps to filter out unrealistic bounding box transformations during data augmentation.
  • 🏎️ These filters aim to improve the stability and effectiveness of model training by avoiding training on poor-quality or highly distorted images that could hamper the learning process.
  • πŸ›  Users can expect potentially more accurate object detection models when these filters are applied during the training of YOLOv5 on custom datasets.

@ml5ah
Copy link
Contributor Author

ml5ah commented May 4, 2021

Hey, @glenn-jocher checking to see if you got a chance to see and had any feedback on this?

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.

3 participants