glenn-jocher commited on
Commit
ad565e3
β€’
1 Parent(s): f80c463

Update README speed reproduction command (#6228)

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -251,7 +251,7 @@ We are super excited about our first-ever Ultralytics YOLOv5 πŸš€ EXPORT Competi
251
 
252
  * All checkpoints are trained to 300 epochs with default settings and hyperparameters.
253
  * **mAP<sup>val</sup>** values are for single-model single-scale on [COCO val2017](http://cocodataset.org) dataset.<br>Reproduce by `python val.py --data coco.yaml --img 640 --conf 0.001 --iou 0.65`
254
- * **Speed** averaged over COCO val images using a [AWS p3.2xlarge](https://aws.amazon.com/ec2/instance-types/p3/) instance. NMS times (~1 ms/img) not included.<br>Reproduce by `python val.py --data coco.yaml --img 640 --conf 0.25 --iou 0.45`
255
  * **TTA** [Test Time Augmentation](https://github.com/ultralytics/yolov5/issues/303) includes reflection and scale augmentations.<br>Reproduce by `python val.py --data coco.yaml --img 1536 --iou 0.7 --augment`
256
 
257
  </details>
 
251
 
252
  * All checkpoints are trained to 300 epochs with default settings and hyperparameters.
253
  * **mAP<sup>val</sup>** values are for single-model single-scale on [COCO val2017](http://cocodataset.org) dataset.<br>Reproduce by `python val.py --data coco.yaml --img 640 --conf 0.001 --iou 0.65`
254
+ * **Speed** averaged over COCO val images using a [AWS p3.2xlarge](https://aws.amazon.com/ec2/instance-types/p3/) instance. NMS times (~1 ms/img) not included.<br>Reproduce by `python val.py --data coco.yaml --img 640 --task speed --batch 1`
255
  * **TTA** [Test Time Augmentation](https://github.com/ultralytics/yolov5/issues/303) includes reflection and scale augmentations.<br>Reproduce by `python val.py --data coco.yaml --img 1536 --iou 0.7 --augment`
256
 
257
  </details>