Skip to content

Commit

Permalink
Update datasets.py
Browse files Browse the repository at this point in the history
  • Loading branch information
glenn-jocher committed Jan 18, 2022
1 parent 29fad3c commit 3a5b9c3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion utils/datasets.py
Original file line number Diff line number Diff line change
Expand Up @@ -983,7 +983,7 @@ def hub_ops(f, max_dim=1920):
data = yaml.safe_load(f) # data dict
if zipped:
data['path'] = data_dir # TODO: should this be dir.resolve()?
check_dataset(data, autodownload=False) # download dataset if missing
check_dataset(data, autodownload=False)
hub_dir = Path(data['path'] + ('-hub' if hub else ''))
stats = {'nc': data['nc'], 'names': data['names']} # statistics dictionary
for split in 'train', 'val', 'test':
Expand Down

0 comments on commit 3a5b9c3

Please sign in to comment.