zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

LoadImagesAndLabels() /images/ to /labels/ fix (#928)

This commit is contained in:
Glenn Jocher 2020-09-10 12:29:05 -07:00
parent d49c52eee3
commit 720645d96a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
utils/datasets.py
View File

@ -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