Bartzi / see

Code for the AAAI 2018 publication "SEE: Towards Semi-Supervised End-to-End Scene Text Recognition"
GNU General Public License v3.0
573 stars 147 forks source link

Prediction Score #70

Open woo1 opened 5 years ago

woo1 commented 5 years ago

I can get predictions, crops, grids data if I predict image. It looks like we can not get any scores. Is it possible to get predicted box score or predicted character score?

Bartzi commented 5 years ago

You can not get any scores for the predicted boxes, at least not with the current implementation (you'd need to add some extra classifiers and think of a way to model a confidence score without knowledge about groundtruth box locations) You can, however, get scores for the characters. Since we are using a softmax classifier at the end, you should be able to extract the maximum value and use this as a confidence score.