TextSnakePostprocessor¶
- class mmocr.models.textdet.TextSnakePostprocessor(text_repr_type='poly', min_text_region_confidence=0.6, min_center_region_confidence=0.2, min_center_area=30, disk_overlap_thr=0.03, radius_shrink_ratio=1.03, rescale_fields=['polygons'], **kwargs)[源代码]¶
Decoding predictions of TextSnake to instances. This was partially adapted from https://github.com/princewang1994/TextSnake.pytorch.
- 参数
text_repr_type (str) – The boundary encoding type ‘poly’ or ‘quad’.
min_text_region_confidence (float) – The confidence threshold of text region in TextSnake.
min_center_region_confidence (float) – The confidence threshold of text center region in TextSnake.
min_center_area (int) – The minimal text center region area.
disk_overlap_thr (float) – The radius overlap threshold for merging disks.
radius_shrink_ratio (float) – The shrink ratio of ordered disks radii.
rescale_fields (list[str], optional) – The bbox/polygon field names to be rescaled. If None, no rescaling will be performed.
- 返回类型
- get_text_instances(pred_results, data_sample)[源代码]¶
- 参数
pred_results (torch.Tensor) – Prediction map with shape \((C, H, W)\).
data_sample (TextDetDataSample) – Datasample of an image.
- 返回
The instance boundary and its confidence.
- 返回类型
- split_results(pred_results)[源代码]¶
Split the prediction results into text score and kernel score.
- 参数
pred_results (torch.Tensor) – The prediction results.
- 返回
The text score and kernel score.
- 返回类型
List[torch.Tensor]