CTCPostProcessor

class mmocr.models.textrecog.CTCPostProcessor(dictionary, max_seq_len=40, ignore_chars=['padding'], **kwargs)

PostProcessor for CTC.

参数

• dictionary (Union[mmocr.models.common.dictionary.dictionary.Dictionary, Dict]) –

• max_seq_len (int) –

• ignore_chars (Sequence[str]) –

返回类型

None

get_single_prediction(probs, data_sample)

Convert the output probabilities of a single image to index and score.

参数

• probs (torch.Tensor) – Character probabilities with shape \((T, C)\).

• data_sample (TextRecogDataSample) – Datasample of an image.

返回

index and score.

返回类型

tuple(list[int], list[float])