text_index type is tensor in converter decode - Githubissues

clovaai / deep-text-recognition-benchmark

Text recognition (optical character recognition) with deep learning methods, ICCV 2019

Apache License 2.0

3.77k stars 1.11k forks source link

text_index type is tensor in converter decode #328

Open JadeKim042386 opened 2 years ago

JadeKim042386 commented 2 years ago

isn't text_index type is tensor here? https://github.com/clovaai/deep-text-recognition-benchmark/blob/68a80fe97943a111ff1efaf52a63ad8f0f1c0e5d/utils.py#L48

When I print the 'preds_index.data' here, it's a tensor. So, above Shouldn't it be an t[i].item()? https://github.com/clovaai/deep-text-recognition-benchmark/blob/68a80fe97943a111ff1efaf52a63ad8f0f1c0e5d/test.py#L118