Benchmark with custom data.yaml (#3858)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Maia Numerosky
2023-07-20 20:44:46 -03:00
committed by GitHub
parent 01dcd54b19
commit aa1cab74f8
3 changed files with 19 additions and 16 deletions

View File

@ -243,7 +243,7 @@ their specific use case based on their requirements for speed and accuracy.
from ultralytics.utils.benchmarks import benchmark
# Benchmark
benchmark(model='yolov8n.pt', imgsz=640, half=False, device=0)
benchmark(model='yolov8n.pt', data='coco8.yaml', imgsz=640, half=False, device=0)
```
[Benchmark Examples](../modes/benchmark.md){ .md-button .md-button--primary}
@ -280,4 +280,4 @@ You can easily customize Trainers to support custom tasks or explore R&D ideas.
Learn more about Customizing `Trainers`, `Validators` and `Predictors` to suit your project needs in the Customization
Section.
[Customization tutorials](engine.md){ .md-button .md-button--primary}
[Customization tutorials](engine.md){ .md-button .md-button--primary}