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

Update HUBDatasetStats() usage examples (#9842)

上级 f1482b06
...@@ -1005,13 +1005,18 @@ def verify_image_label(args): ...@@ -1005,13 +1005,18 @@ def verify_image_label(args):
class HUBDatasetStats(): class HUBDatasetStats():
""" Return dataset statistics dictionary with images and instances counts per split per class """ Class for generating HUB dataset JSON and `-hub` dataset directory
To run in parent directory: export PYTHONPATH="$PWD/yolov5"
Usage1: from utils.dataloaders import *; HUBDatasetStats('coco128.yaml', autodownload=True)
Usage2: from utils.dataloaders import *; HUBDatasetStats('path/to/coco128_with_yaml.zip')
Arguments Arguments
path: Path to data.yaml or data.zip (with data.yaml inside data.zip) path: Path to data.yaml or data.zip (with data.yaml inside data.zip)
autodownload: Attempt to download dataset if not found locally autodownload: Attempt to download dataset if not found locally
Usage
from utils.dataloaders import HUBDatasetStats
stats = HUBDatasetStats('coco128.yaml', autodownload=True) # usage 1
stats = HUBDatasetStats('path/to/coco128.zip') # usage 2
stats.get_json(save=False)
stats.process_images()
""" """
def __init__(self, path='coco128.yaml', autodownload=False): def __init__(self, path='coco128.yaml', autodownload=False):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论