towhee.trainer.utils.plot_utils.predict_image_classification¶
- towhee.trainer.utils.plot_utils.predict_image_classification(model: Module, input_: Tensor)[source]¶
Predict using an image classification model. :param model: Pytorch model. :type model: nn.Module :param input_: Input image tensor. :type input_: Tensor
- Returns:
- (tuple)
Prediction score which max is 1, and label idx.