Skip to content

Commit

Permalink
Remove dataset_stats() autodownload capability (ultralytics#6303)
Browse files Browse the repository at this point in the history
* Remove `dataset_stats()` autodownload capability

@kalenmike security update per Slack convo

* Update datasets.py
  • Loading branch information
glenn-jocher authored Jan 18, 2022
1 parent f62ba67 commit 5aa3534
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions utils/datasets.py
Original file line number Diff line number Diff line change
Expand Up @@ -936,11 +936,10 @@ def verify_image_label(args):
return [None, None, None, None, nm, nf, ne, nc, msg]


def dataset_stats(path='coco128.yaml', autodownload=False, verbose=False, profile=False, hub=False):
def dataset_stats(path='coco128.yaml', verbose=False, profile=False, hub=False):
""" Return dataset statistics dictionary with images and instances counts per split per class
To run in parent directory: export PYTHONPATH="$PWD/yolov5"
Usage1: from utils.datasets import *; dataset_stats('coco128.yaml', autodownload=True)
Usage2: from utils.datasets import *; dataset_stats('../datasets/coco128_with_yaml.zip')
Usage: from utils.datasets import *; dataset_stats('../datasets/coco128_with_yaml.zip')
Arguments
path: Path to data.yaml or data.zip (with data.yaml inside data.zip)
autodownload: Attempt to download dataset if not found locally
Expand Down Expand Up @@ -984,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) # 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 5aa3534

Please sign in to comment.