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_pathandann_pathare the same and contain.txtfiles 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.