Explorar o código

fix: adjust condition for block count to enhance detection logic

myhloli hai 3 meses
pai
achega
34360ba642
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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)