mmcls.apis.show_result_pyplot

mmcls.apis.show_result_pyplot(model, img, result, fig_size=(15, 10), title='result', wait_time=0)

Visualize the classification results on the image.

Parameters

• model (nn.Module) – The loaded classifier.

• img (str or np.ndarray) – Image filename or loaded image.

• result (list) – The classification result.

• fig_size (tuple) – Figure size of the pyplot figure. Defaults to (15, 10).

• title (str) – Title of the pyplot figure. Defaults to ‘result’.

• wait_time (int) – How many seconds to display the image. Defaults to 0.