self.label_files bug fix (#958)

ultralytics · Sep 13, 2020 · 806e75f · 806e75f
1 parent 6062bb7
commit 806e75f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/datasets.py b/utils/datasets.py
@@ -364,7 +364,7 @@ def __init__(self, path, img_size=640, batch_size=16, augment=False, hyp=None, r
 
         # 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