Improve classification comments (#8997)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-08-17 15:33:37 +02:00
1 parent d3ea0df8b9
commit e617569107
5 files changed
+16 -13

No files matched your search

+2 -1
View File
@@ -3,7 +3,8 @@
Validate a classification model on a dataset
Usage:
$ python classify/val.py --weights yolov5s-cls.pt --data ../datasets/imagenet
$ bash data/scripts/get_imagenet.sh --val # download ImageNet val split (6.3G, 50000 images)
$ python classify/val.py --weights yolov5m-cls.pt --data ../datasets/imagenet --img 224 # validate
"""
import argparse