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

Resolve dataset paths (#6489)

This commit is contained in:
Glenn Jocher 2022-02-01 15:54:51 +01:00 committed by GitHub
parent 7539cd75c3
commit 6445a8137e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
utils/general.py
View File

@ -394,12 +394,15 @@ def check_dataset(data, autodownload=True):
with open(data, errors='ignore') as f: with open(data, errors='ignore') as f:
data = yaml.safe_load(f) # dictionary data = yaml.safe_load(f) # dictionary
# Parse yaml # Resolve paths
path = extract_dir or Path(data.get('path') or '') # optional 'path' default to '.' path = Path(extract_dir or data.get('path') or '') # optional 'path' default to '.'
if not path.is_absolute():
path = (ROOT / path).resolve()
for k in 'train', 'val', 'test': for k in 'train', 'val', 'test':
if data.get(k): # prepend path 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]] data[k] = str(path / data[k]) if isinstance(data[k], str) else [str(path / x) for x in data[k]]
# Parse yaml
assert 'nc' in data, "Dataset 'nc' key missing." assert 'nc' in data, "Dataset 'nc' key missing."
if 'names' not in data: if 'names' not in data:
data['names'] = [f'class{i}' for i in range(data['nc'])] # assign class names if missing data['names'] = [f'class{i}' for i in range(data['nc'])] # assign class names if missing