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

Author: xinke-wang

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()

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