Unverified 提交 b83e1a4a authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Fix `img2label_paths()` order (#3720)

* Fix `img2label_paths()` order * fix, 1
上级 1f69d125
......@@ -348,7 +348,7 @@ class LoadStreams: # multiple IP or RTSP cameras
def img2label_paths(img_paths):
# Define label paths as a function of image paths
sa, sb = os.sep + 'images' + os.sep, os.sep + 'labels' + os.sep # /images/, /labels/ substrings
return ['txt'.join(x.replace(sa, sb, 1).rsplit(x.split('.')[-1], 1)) for x in img_paths]
return [sb.join(x.rsplit(sa, 1)).rsplit('.', 1)[0] + '.txt' for x in img_paths]
class LoadImagesAndLabels(Dataset): # for training/testing
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论