self.label_files bug fix (#958)

This commit is contained in:
Glenn Jocher committed 2020-09-13 11:38:23 -07:00
1 parent 6062bb7dee
commit 806e75f2b1
1 file changed
+1 -1
+1 -1
View File
@@ -364,7 +364,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
# Define labels
sa, sb = os.sep + 'images' + os.sep, os.sep + 'labels' + os.sep # /images/, /labels/ substrings
self.label_files = [sb.join(x.rsplit(sa, 1)).replace(os.path.splitext(x)[-1], '.txt') for x in self.img_files]
self.label_files = [x.replace(sa, sb, 1).replace(os.path.splitext(x)[-1], '.txt') for x in self.img_files]
# Check cache
cache_path = str(Path(self.label_files[0]).parent) + '.cache' # cached labels