OrgLion-ML
/
deep-text-recognition-benchmark
mirror of https://github.com/clovaai/deep-text-recognition-benchmark.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Correct demo.py error when running with CTC

This commit is contained in:
rhcsky 2020-07-24 10:25:30 +09:00
parent 425a9b3e1f
commit 7f59fb000f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
demo.py
View File

@ -57,7 +57,7 @@ def demo(opt):
# Select max probabilty (greedy decoding) then decode index to character # Select max probabilty (greedy decoding) then decode index to character
preds_size = torch.IntTensor([preds.size(1)] * batch_size) preds_size = torch.IntTensor([preds.size(1)] * batch_size)
_, preds_index = preds.max(2) _, preds_index = preds.max(2)
preds_index = preds_index.view(-1) # preds_index = preds_index.view(-1)
preds_str = converter.decode(preds_index.data, preds_size.data) preds_str = converter.decode(preds_index.data, preds_size.data)
else: else: