ultralytics/docs/modes/benchmark.md

Benchmark mode is used to profile the speed and accuracy of various export formats for YOLOv8. The benchmarks provide information on the size of the exported format, its mAP50-95 metrics (for object detection and segmentation) or accuracy_top5 metrics (for classification), and the inference time in milliseconds per image across various export formats like ONNX, OpenVINO, TensorRT and others. This information can help users choose the optimal export format for their specific use case based on their requirements for speed and accuracy.

!!! tip "Tip"

* Export to ONNX or OpenVINO for up to 3x CPU speedup.
* Export to TensorRT for up to 5x GPU speedup.

Usage Examples

Run YOLOv8n benchmarks on all supported export formats including ONNX, TensorRT etc. See Arguments section below for a full list of export arguments.

!!! example ""

=== "Python"

    ```python
    from ultralytics.yolo.utils.benchmarks import benchmark
    
    # Benchmark
    benchmark(model='yolov8n.pt', imgsz=640, half=False, device=0)
    ```
=== "CLI"

    ```bash
    yolo benchmark model=yolov8n.pt imgsz=640 half=False device=0
    ```

Arguments

Arguments such as model, imgsz, half, device, and hard_fail provide users with the flexibility to fine-tune the benchmarks to their specific needs and compare the performance of different export formats with ease.

Key	Value	Description
model	None	path to model file, i.e. yolov8n.pt, yolov8n.yaml
imgsz	640	image size as scalar or (h, w) list, i.e. (640, 480)
half	False	FP16 quantization
device	None	device to run on, i.e. cuda device=0 or device=0,1,2,3 or device=cpu
hard_fail	False	do not continue on error (bool), or val floor threshold (float)

Export Formats

Benchmarks will attempt to run automatically on all possible export formats below.

Format	format Argument	Model	Metadata
PyTorch	-	yolov8n.pt	✅
TorchScript	torchscript	yolov8n.torchscript	✅
ONNX	onnx	yolov8n.onnx	✅
OpenVINO	openvino	yolov8n_openvino_model/	✅
TensorRT	engine	yolov8n.engine	✅
CoreML	coreml	yolov8n.mlmodel	✅
TF SavedModel	saved_model	yolov8n_saved_model/	✅
TF GraphDef	pb	yolov8n.pb	❌
TF Lite	tflite	yolov8n.tflite	✅
TF Edge TPU	edgetpu	yolov8n_edgetpu.tflite	✅
TF.js	tfjs	yolov8n_web_model/	✅
PaddlePaddle	paddle	yolov8n_paddle_model/	✅