Evaluator

class mmengine.evaluator.Evaluator(metrics)

Wrapper class to compose multiple BaseMetric instances.

参数:

metrics (dict or BaseMetric or Sequence) – The config of metrics.

property dataset_meta: dict | None

Meta info of the dataset.

Type:

Optional[dict]

evaluate(size)

Invoke evaluate method of each metric and collect the metrics dictionary.

参数:

size (int) – Length of the entire validation dataset. When batch size > 1, the dataloader may pad some data samples to make sure all ranks have the same length of dataset slice. The collect_results function will drop the padded data based on this size.

返回:

Evaluation results of all metrics. The keys are the names of the metrics, and the values are corresponding results.

返回类型:

dict

offline_evaluate(data_samples, data=None, chunk_size=1)

Offline evaluate the dumped predictions on the given data .

参数:

• data_samples (Sequence) – All predictions and ground truth of the model and the validation set.

• data (Sequence, optional) – All data of the validation set.

• chunk_size (int) – The number of data samples and predictions to be processed in a batch.

process(data_samples, data_batch=None)

Convert BaseDataSample to dict and invoke process method of each metric.

参数:

• data_samples (Sequence[BaseDataElement]) – predictions of the model, and the ground truth of the validation set.

• data_batch (Any, optional) – A batch of data from the dataloader.