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

Common is_coco logic betwen train.py and val.py #5521

Merged
merged 1 commit into from
Nov 5, 2021
Merged

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Nov 5, 2021

Resolves #3644

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Update to COCO dataset detection logic in training script.

πŸ“Š Key Changes

  • Adjusted the method to identify if the COCO dataset is being used.
  • Previously, checked if the data configuration ended with 'coco.yaml' and had 80 classes.
  • Now, checks if the val_path is a string that ends with 'coco/val2017.txt'.

🎯 Purpose & Impact

  • 🎨 Ensures more accurate detection of COCO dataset use by examining validation path.
  • βš™οΈ Potentially impacts users training models with COCO dataset, improving reliability for model setup.
  • πŸš€ Users can expect a more robust way of identifying the usage of COCO dataset, leading to fewer errors and assumptions during training.

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.

Training crash on first validation: Error in processing validate dataset path
1 participant