提交 720645d9 authored 作者: Glenn Jocher's avatar Glenn Jocher

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

上级 d49c52ee
......@@ -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
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论