yolov5/utils
Glenn Jocher 7a565f130a
Update `dataset_stats()` (#3593)
@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)
```
2021-06-12 13:26:41 +02:00
..
aws Explicit `git clone` master (#3311) 2021-05-24 12:42:36 +02:00
flask_rest_api Update JSON response (#3139) 2021-05-12 18:23:50 +02:00
google_app_engine Bump pip from 18.1 to 19.2 in /utils/google_app_engine (#3561) 2021-06-09 21:14:09 +02:00
wandb_logging Improves docs and handling of entities and resuming by WandbLogger (#3264) 2021-05-21 23:42:53 +02:00
__init__.py initial commit 2020-05-29 17:04:54 -07:00
activations.py ACON Activation batch-size 1 bug patch (#2901) 2021-04-25 22:54:07 +02:00
autoanchor.py Update check_requirements() exclude list (#2974) 2021-04-29 21:16:23 +02:00
datasets.py Update `dataset_stats()` (#3593) 2021-06-12 13:26:41 +02:00
general.py Update `check_python(minimum=3.6.2)` (#3548) 2021-06-09 11:34:15 +02:00
google_utils.py Parse URL authentication (#3424) 2021-06-03 18:44:58 +02:00
loss.py Add --label-smoothing eps argument to train.py (default 0.0) (#2344) 2021-03-29 18:45:46 +02:00
metrics.py ConfusionMatrix `normalize=True` fix (#3587) 2021-06-11 11:46:05 +02:00
plots.py Ignore Seaborn plot warnings (#3576) 2021-06-10 13:51:29 +02:00
torch_utils.py Multi-GPU default to single device 0 (#3554) 2021-06-09 15:31:47 +02:00