attention-ocr
attention-ocr copied to clipboard
Return multiple guesses and probabilities for a given image
As briefly discussed in #19, it would be great to return a list of "guesses" with their probabilities. e.g. for an image like "JABRONI" we could get a response a la
[
{
"output": "JABRONI",
"probability": 0.998
},
{
"output": "JABR0NI",
"probability": 0.968
},
{
"output": "JABR0N1",
"probability": 0.942
}
]
adding this here to track it. i may find some time to get to it, but not in the short term. think we just need a clever way to go through the probabilities and determine which guesses we should consider. somewhere around here: https://github.com/emedvedev/attention-ocr/blob/45052281b9a907eb888c252ec4c821171b1808d2/aocr/model/model.py#L177-L179
@emedveded Any ideas on how to do this ?
I could try to work on it because I need this functionality but I have first to understand precisely how the prediction is done