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

🐛[Fix] fix validator to handle custom datasets with txt files #95

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

DizzyZff
Copy link

@DizzyZff DizzyZff commented Aug 29, 2024

Description

When the input validation dataset is using txt labels the Validator throws exception on these lines:

https://github.com/WongKinYiu/YOLO/blob/8228669808a626fc5f9c233fdb35550b5e041fae/yolo/tools/solver.py#L231-L233

Add a condition for txt files to bypass this if.

Closes #94

Type of Change

Please delete options that are not relevant.

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to not work as expected)
  • This change requires a documentation update

Checklist:

  • The code follows the Python style guide.
  • Code and files are well organized.
  • All tests pass.
  • New code is covered by tests.
  • The pull request is directed to the corresponding topic branch.
  • [Optional] We would be very happy if gitmoji 🧑‍💻 could be used to assist the commit message 💬!

Licensing:

By submitting this pull request, I confirm that:

  • My contribution is made under the MIT License.
  • I have not included any code from questionable or non-compliant sources (GPL, AGPL, ... etc).
  • I understand that all contributions to this repository must comply with the MIT License, and I promise that my contributions do not violate this license.
  • I have not used any code or content from sources that conflict with the MIT License or are otherwise legally questionable.

@DizzyZff DizzyZff changed the title [Fix] Fix validator to handle custom datasets with txt files 🐛[Fix] fix validator to handle custom datasets with txt files Aug 29, 2024
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.

Problem with validator when training with Custom Dataset with txt files
3 participants