data.yaml names.keys() integer assert (#10190)
* data.yaml `names.keys()` integer assert Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Update general.py Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
1 parent
a9f895d304
commit
1510111b46
1 file changed
+2
-1
+2
-1
@@ -482,9 +482,10 @@ def check_dataset(data, autodownload=True):
|
||||
|
||||
# Checks
|
||||
for k in 'train', 'val', 'names':
|
||||
assert k in data, f"data.yaml '{k}:' field missing ❌"
|
||||
assert k in data, emojis(f"data.yaml '{k}:' field missing ❌")
|
||||
if isinstance(data['names'], (list, tuple)): # old array format
|
||||
data['names'] = dict(enumerate(data['names'])) # convert to dict
|
||||
assert all(isinstance(k, int) for k in data['names'].keys()), 'data.yaml names keys must be integers, i.e. 2: car'
|
||||
data['nc'] = len(data['names'])
|
||||
|
||||
# Resolve paths
|
||||
|
||||
Reference in new issue
Block a user