Merge pull request #62 from myhloli/master

update confidence score 0.95->0.05
opendatalab · Apr 23, 2024 · fc23483 · fc23483
2 parents ec2cbdd + 1146206
commit fc23483
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/magic_pdf/model/magic_model.py b/magic_pdf/model/magic_model.py
@@ -48,7 +48,7 @@ def __fix_by_confidence(self):
             need_remove_list = []
             layout_dets = model_page_info["layout_dets"]
             for layout_det in layout_dets:
-                if layout_det["score"] < 0.95:
+                if layout_det["score"] <= 0.05:
                     need_remove_list.append(layout_det)
                 else:
                     continue

diff --git a/magic_pdf/pre_proc/ocr_dict_merge.py b/magic_pdf/pre_proc/ocr_dict_merge.py
@@ -120,14 +120,14 @@ def sort_blocks_by_layout(all_bboxes, layout_bboxes):
             # 如果是footnote则跳过
             if block[7] == BlockType.Footnote:
                 continue
-            block_bbox = [block[0], block[1], block[2], block[3]]
+            block_bbox = block[:4]
             if calculate_overlap_area_in_bbox1_area_ratio(block_bbox, layout_bbox) > 0.8:
                 layout_blocks.append(block)
 
         # 如果layout_blocks不为空，则放入new_blocks中
         if len(layout_blocks) > 0:
             new_blocks.append(layout_blocks)
-            # 从spans删除已经放入layout_sapns中的span
+            # 从all_bboxes删除已经放入layout_blocks中的block
             for layout_block in layout_blocks:
                 all_bboxes.remove(layout_block)