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 yolov5 yaml build to handle case where directory with yaml is provided #266

Merged
merged 1 commit into from
Jul 19, 2023

Conversation

dsikka
Copy link
Contributor

@dsikka dsikka commented Jul 19, 2023

For this ticket: https://app.asana.com/0/1204322589046915/1205045628929361/f

Previously we handled 2 cases:

  • User provides path to a yaml
  • User provides a directory of custom data + classes and we build the yaml for them

This adds the third case where the user provides a directory and we first check if a yaml already exists in the folder. If there is a yaml, we use that as the data.

Tested locally using a training-aware run

@dsikka dsikka marked this pull request as ready for review July 19, 2023 13:43
@dsikka dsikka merged commit 532a755 into main Jul 19, 2023
3 checks passed
@dsikka dsikka deleted the update_yolov5_yaml branch July 19, 2023 14:12
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