Shortcuts

mmcls.apis.single_gpu_test

mmcls.apis.single_gpu_test(model, data_loader, show=False, out_dir=None, **show_kwargs)[源代码]

Test model with local single gpu.

This method tests model with a single gpu and supports showing results.

参数
  • model (torch.nn.Module) – Model to be tested.

  • data_loader (torch.utils.data.DataLoader) – Pytorch data loader.

  • show (bool) – Whether to show the test results. Defaults to False.

  • out_dir (str) – The output directory of result plots of all samples. Defaults to None, which means not to write output files.

  • **show_kwargs – Any other keyword arguments for showing results.

返回

The prediction results.

返回类型

list

Read the Docs v: latest
Versions
master
latest
1.x
dev-1.x
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.