Parcourir la source

fix: adjust threshold for wired model detection based on wireless model results

myhloli il y a 3 mois
Parent
commit
a086cfad0d
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      mineru/model/table/rec/unet_table/main.py

+ 1 - 1
mineru/model/table/rec/unet_table/main.py

@@ -297,7 +297,7 @@ class UnetTableModel:
                 gap_of_len = wireless_len - wired_len
                 # 判断是否使用无线表格模型的结果
                 if (
-                    wired_len <= round(wireless_len * 0.6)  # 有线模型检测到的单元格数太少(低于无线模型的50%)
+                    wired_len <= int(wireless_len * 0.55)+1  # 有线模型检测到的单元格数太少(低于无线模型的50%)
                     # or ((round(wireless_len*1.2) < wired_len) and (wired_len < (2 * wireless_len)) and table_cls_score <= 0.94)  # 有线模型检测到的单元格数反而更多
                     or (0 <= gap_of_len <= 5 and wired_len <= round(wireless_len * 0.75))  # 两者相差不大但有线模型结果较少
                     or (gap_of_len == 0 and wired_len <= 4)  # 单元格数量完全相等且总量小于等于4