Commit
·
720645d
1
Parent(s):
d49c52e
LoadImagesAndLabels() /images/ to /labels/ fix (#928)
Browse files- utils/datasets.py +2 -2
utils/datasets.py
CHANGED
@@ -363,8 +363,8 @@ class LoadImagesAndLabels(Dataset): # for training/testing
|
|
363 |
self.stride = stride
|
364 |
|
365 |
# Define labels
|
366 |
-
|
367 |
-
|
368 |
|
369 |
# Check cache
|
370 |
cache_path = str(Path(self.label_files[0]).parent) + '.cache' # cached labels
|
|
|
363 |
self.stride = stride
|
364 |
|
365 |
# Define labels
|
366 |
+
sa, sb = os.sep + 'images' + os.sep, os.sep + 'labels' + os.sep # /images/, /labels/ substrings
|
367 |
+
self.label_files = [sb.join(x.rsplit(sa, 1)).replace(os.path.splitext(x)[-1], '.txt') for x in self.img_files]
|
368 |
|
369 |
# Check cache
|
370 |
cache_path = str(Path(self.label_files[0]).parent) + '.cache' # cached labels
|