Skip to content

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

Merged
glenn-jocher merged 1 commit intomasterfrom
update/is_coco
Nov 5, 2021
Merged

Common is_coco logic betwen train.py and val.py#5521
glenn-jocher merged 1 commit intomasterfrom
update/is_coco

Conversation

@glenn-jocher
Copy link
Copy Markdown
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