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

Missing `nc` and `names` handling in check_dataset() (#4066)

上级 b1be6850
......@@ -231,6 +231,9 @@ def check_dataset(data, autodownload=True):
if data.get(k): # prepend path
data[k] = str(path / data[k]) if isinstance(data[k], str) else [str(path / x) for x in data[k]]
assert 'nc' in data, "Dataset 'nc' key missing."
if 'names' not in data:
data['names'] = [str(i) for i in range(data['nc'])] # assign class names if missing
train, val, test, s = [data.get(x) for x in ('train', 'val', 'test', 'download')]
if val:
val = [Path(x).resolve() for x in (val if isinstance(val, list) else [val])] # val path
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论