mmpretrain.apis.inference_model¶
- mmpretrain.apis.inference_model(model, *args, **kwargs)[源代码]¶
Inference an image with the inferencer.
Automatically select inferencer to inference according to the type of model. It’s a shortcut for a quick start, and for advanced usage, please use the correspondding inferencer class.
Here is the mapping from task to inferencer:
Image Classification:
ImageClassificationInferencer
Image Retrieval:
ImageRetrievalInferencer
Image Caption:
ImageCaptionInferencer
Visual Question Answering:
VisualQuestionAnsweringInferencer
Visual Grounding:
VisualGroundingInferencer
Text-To-Image Retrieval:
TextToImageRetrievalInferencer
Image-To-Text Retrieval:
ImageToTextRetrievalInferencer
NLVR:
NLVRInferencer