-
Notifications
You must be signed in to change notification settings - Fork 1.1k
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
Implemented yolo dataset support #487
Conversation
Thanks for your contribution! Would you like to add a brief introduction to the Yolo format dataset in the README? Line 218 in d8ba391
Also, please run |
Thank you for the hint about the formatting, I have updated the code using the |
@RangiLyu Readme has been adjusted as well. Would be great if you could have a look again. |
Codecov Report
@@ Coverage Diff @@
## main #487 +/- ##
==========================================
- Coverage 74.65% 73.60% -1.05%
==========================================
Files 70 71 +1
Lines 4600 4691 +91
Branches 716 728 +12
==========================================
+ Hits 3434 3453 +19
- Misses 975 1045 +70
- Partials 191 193 +2
Flags with carried forward coverage won't be shown. Click here to find out more.
📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more |
I have implemented yolo dataset support to load data in the YOLO format as mentioned in #10. I have used the existing XML-to-COCO dataset and adapted it to support the yolo format.
The loader expects the image files to be either
".png", ".jpg", ".jpeg", ".bmp", ".tiff"
and tries to find the best-matching one. Usually theimg_path
andann_path
are the same and contain.txt
files as well as the image files.Loading YOLO datasets can be slower because we have to open every image to determine the image size. This could be enhanced by using multi-processing, which currently is not implemented.