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

Update ClassificationModel default training imgsz=224 #9469

Merged
merged 1 commit into from
Sep 18, 2022

Conversation

glenn-jocher
Copy link
Member

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

To match classify/val.py and classify/predict.py

Helps #9462

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

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

YOLOv5 classification model now trains with higher resolution images by default.

πŸ“Š Key Changes

  • The default input image size for training has been changed from 128 to 224 pixels.
  • The example usage in the documentation has been updated to reflect this new default image size.

🎯 Purpose & Impact

  • 🎨 Enhances the model's ability to learn from higher resolution images, which can potentially improve classification accuracy.
  • 🧠 May impact computational requirements, as higher resolution images consume more memory and processing power.
  • πŸ“ˆ Users should see an improvement in the model performance on classification tasks with adequate hardware.

To match classify/val.py and classify/predict.py

Helps #9462

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
@glenn-jocher glenn-jocher self-assigned this Sep 18, 2022
@glenn-jocher glenn-jocher merged commit 295c5e9 into master Sep 18, 2022
@glenn-jocher glenn-jocher deleted the glenn-jocher-patch-1 branch September 18, 2022 14:13
@glenn-jocher glenn-jocher linked an issue Sep 18, 2022 that may be closed by this pull request
1 task
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.

The recognition result of classify/predict.py is very poor?
1 participant