Shortcuts

Note

You are reading the documentation for MMClassification 0.x, which will soon be deprecated at the end of 2022. We recommend you upgrade to MMClassification 1.0 to enjoy fruitful new features and better performance brought by OpenMMLab 2.0. Check the installation tutorial, migration tutorial and changelog for more details.

mmcls.apis.single_gpu_test

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

Test model with local single gpu.

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

Parameters
  • 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.

Returns

The prediction results.

Return type

list

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

Free document hosting provided by Read the Docs.