Jelajahi Sumber

fix: adjust condition for block count to enhance detection logic

myhloli 3 bulan lalu
induk
melakukan
34360ba642
1 mengubah file dengan 1 tambahan dan 1 penghapusan
  1. 1 1
      mineru/utils/model_utils.py

+ 1 - 1
mineru/utils/model_utils.py

@@ -290,7 +290,7 @@ def remove_overlaps_low_confidence_blocks(combined_res_list, overlap_threshold=0
                                                                              overlap_threshold)]
 
         # 如果内部有3个及以上的小block
-        if len(blocks_inside) >= 3:
+        if len(blocks_inside) >= 2:
             # 计算小block的平均分数
             avg_score = sum(s for _, s, _ in blocks_inside) / len(blocks_inside)