yolo.model.metrics

Attributes

YOLO_METRICS_MAP

Classes

Metrics	Class for computing metrics.
MetricParser	Class for parsing Ultralytics metrics in results.csv.

Module Contents

yolo.model.metrics.YOLO_METRICS_MAP

class yolo.model.metrics.Metrics(task: str, results: ultralytics.engine.results, metrics_list: list)

Class for computing metrics.

_TASKS_AVAILABLE

_METRICS_AVAILABLE

task

results

metrics_list

metric_dict

static metrics_task(results: ultralytics.engine.results, task: str)

Return the task metrics to be computes.

Parameters:

• results (r) – Results as defined by Ultralytics.

• task (str) – segment or detect or obb.

Returns:

results.seg for segment or results.box for detect or obb.

compute_metrics()

Compute all metrics contained in parameter.

_compute_metric(metric: str)

Check if the metric is available and return the associated yolo metric.

class yolo.model.metrics.MetricParser(filepath: str)

Class for parsing Ultralytics metrics in results.csv.

df

log_metrics()

Log the metrics inside the csv file.