[Fix] Fix Recognition Score Normalization Issue (#1333)

open-mmlab · Sep 21, 2022 · 0421659 · 0421659
1 parent 3c63f73
commit 0421659
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/mmocr/models/textrecog/convertors/attn.py b/mmocr/models/textrecog/convertors/attn.py
@@ -123,7 +123,6 @@ def tensor2idx(self, outputs, img_metas=None):
         indexes, scores = [], []
         for idx in range(batch_size):
             seq = outputs[idx, :, :]
-            seq = seq.softmax(dim=-1)
             max_value, max_idx = torch.max(seq, -1)
             str_index, str_score = [], []
             output_index = max_idx.cpu().detach().numpy().tolist()

diff --git a/mmocr/models/textrecog/recognizer/abinet.py b/mmocr/models/textrecog/recognizer/abinet.py
@@ -181,7 +181,7 @@ def simple_test(self, img, img_metas, **kwargs):
             return ret['logits']
 
         label_indexes, label_scores = self.label_convertor.tensor2idx(
-            ret['logits'], img_metas)
+            ret['logits'].softmax(dim=-1), img_metas)
         label_strings = self.label_convertor.idx2str(label_indexes)
 
         # flatten batch results