glenn-jocher commited on
Commit
806e75f
·
1 Parent(s): 6062bb7

self.label_files bug fix (#958)

Browse files
Files changed (1) hide show
  1. utils/datasets.py +1 -1
utils/datasets.py CHANGED
@@ -364,7 +364,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
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
 
364
 
365
  # Define labels
366
  sa, sb = os.sep + 'images' + os.sep, os.sep + 'labels' + os.sep # /images/, /labels/ substrings
367
+ self.label_files = [x.replace(sa, sb, 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