• Glenn Jocher's avatar
    Update `dataset_stats()` (#3593) · 7a565f13
    Glenn Jocher 提交于
    @KalenMike this is a PR to add image filenames and labels to our stats dictionary and to save the dictionary to JSON. Save location is next to the train labels.cache file. The single JSON contains all stats for entire dataset.
    
    Usage example:
    ```python
    from utils.datasets import *
    
    dataset_stats('coco128.yaml', verbose=True)
    ```
    7a565f13
datasets.py 46.5 KB