diff --git a/utils/datasets.py b/utils/datasets.py index a5396f0..d61b86c 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -363,8 +363,8 @@ class LoadImagesAndLabels(Dataset): # for training/testing self.stride = stride # Define labels - self.label_files = [x.replace('images', 'labels').replace(os.path.splitext(x)[-1], '.txt') for x in - self.img_files] + 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] # Check cache cache_path = str(Path(self.label_files[0]).parent) + '.cache' # cached labels