DumpResults

class mmengine.evaluator.DumpResults(out_file_path, collect_device='cpu', collect_dir=None)

Dump model predictions to a pickle file for offline evaluation.

参数:

• out_file_path (str) – Path of the dumped file. Must end with ‘.pkl’ or ‘.pickle’.

• collect_device (str) – Device name used for collecting results from different ranks during distributed training. Must be ‘cpu’ or ‘gpu’. Defaults to ‘cpu’.

• collect_dir (str | None) – (str, optional): Synchronize directory for collecting data from different ranks. This argument should only be configured when collect_device is ‘cpu’. Defaults to None. New in version 0.7.3.

compute_metrics(results)

Dump the prediction results to a pickle file.

参数:

results (list) –

返回类型:

dict

process(data_batch, predictions)

Transfer tensors in predictions to CPU.

参数:

• data_batch (Any) –

• predictions (Sequence[dict]) –

返回类型:

None