search

open-mmlab / mmocr

OpenMMLab Text Detection, Recognition and Understanding Toolbox
https://mmocr.readthedocs.io/en/dev-1.x/
Apache License 2.0
4.29k stars 744 forks source link

[Feature] Is it possible to show confidence scores for text instances when visualizing test results? #1692

Open OstrichThinking opened 1 year ago

OstrichThinking commented 1 year ago

What is the feature?

The confidence scores of detection results can help us analyze the performance of the model.

Any other context?

No response

Harold-lkk commented 1 year ago

Current visualization can not show confidence scores. You can implement it at https://github.com/open-mmlab/mmocr/blob/689ecf0f5fca20975131e0c30552ebbfe526bddb/mmocr/visualization/textdet_visualizer.py#L66 by adding score parameter and drawing score text