Shortcuts

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.

参数:
返回类型:

None

Read the Docs v: latest
Versions
latest
stable
v0.10.3
v0.10.2
v0.10.1
v0.10.0
v0.9.1
v0.9.0
v0.8.5
v0.8.4
v0.8.3
v0.8.2
v0.8.1
v0.8.0
v0.7.4
v0.7.3
v0.7.2
v0.7.1
v0.7.0
v0.6.0
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.