TextDetInferencer¶
- class mmocr.apis.inferencers.TextDetInferencer(model=None, weights=None, device=None, scope='mmocr')[源代码]¶
Text Detection inferencer.
- 参数
model (str, optional) – Path to the config file or the model name defined in metafile. For example, it could be “dbnet_resnet18_fpnc_1200e_icdar2015” or “configs/textdet/dbnet/dbnet_resnet18_fpnc_1200e_icdar2015.py”. If model is not specified, user must provide the weights saved by MMEngine which contains the config string. Defaults to None.
weights (str, optional) – Path to the checkpoint. If it is not specified and model is a model name of metafile, the weights will be loaded from metafile. Defaults to None.
device (str, optional) – Device to run inference. If None, the available device will be automatically used. Defaults to None.
scope (str, optional) – The scope of the model. Defaults to “mmocr”.
- 返回类型
- pred2dict(data_sample)[源代码]¶
Extract elements necessary to represent a prediction into a dictionary. It’s better to contain only basic data elements such as strings and numbers in order to guarantee it’s json-serializable.
- 参数
data_sample (TextDetDataSample) – The data sample to be converted.
- 返回
The output dictionary.
- 返回类型