fix(python): Modify the way CTCLabelDecode calculates confidence (#179)

RapidAI · May 18, 2024 · 41a23c0 · 41a23c0
1 parent 0e1cb18
commit 41a23c0
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/python/rapidocr_onnxruntime/ch_ppocr_v3_rec/utils.py b/python/rapidocr_onnxruntime/ch_ppocr_v3_rec/utils.py
@@ -69,5 +69,5 @@ def decode(self, text_index, text_prob=None, is_remove_duplicate=False):
                 else:
                     conf_list.append(1)
             text = "".join(char_list)
-            result_list.append((text, np.mean(conf_list + [1e-50])))
+            result_list.append((text, np.mean(conf_list if any(conf_list) else [0])))
         return result_list
diff --git a/python/rapidocr_openvino/ch_ppocr_v3_rec/utils.py b/python/rapidocr_openvino/ch_ppocr_v3_rec/utils.py
@@ -67,5 +67,5 @@ def decode(self, text_index, text_prob=None, is_remove_duplicate=False):
                 else:
                     conf_list.append(1)
             text = "".join(char_list)
-            result_list.append((text, np.mean(conf_list + [1e-50])))
+            result_list.append((text, np.mean(conf_list if any(conf_list) else [0])))
         return result_list
diff --git a/python/rapidocr_paddle/ch_ppocr_v3_rec/utils.py b/python/rapidocr_paddle/ch_ppocr_v3_rec/utils.py
@@ -71,5 +71,5 @@ def decode(self, text_index, text_prob=None, is_remove_duplicate=False):
                 else:
                     conf_list.append(1)
             text = "".join(char_list)
-            result_list.append((text, np.mean(conf_list + [1e-50])))
+            result_list.append((text, np.mean(conf_list if any(conf_list) else [0])))
         return result_list