changdazhou 8 місяців тому
батько
коміт
f94ef15df8
1 змінених файлів з 10 додано та 11 видалено
  1. 10 11
      paddlex/inference/pipelines/layout_parsing/result_v2.py

+ 10 - 11
paddlex/inference/pipelines/layout_parsing/result_v2.py

@@ -332,31 +332,30 @@ class LayoutParsingResultV2(BaseCVResult, HtmlMixin, XlsxMixin, MarkdownMixin):
                     prev_block_bbox = prev_block["block_bbox"]
                     num_of_prev_lines = prev_block.get("num_of_lines")
                     pre_block_seg_end_coordinate = prev_block.get("seg_end_coordinate")
+                    prev_end_space_small = (
+                        context_right_coordinate - pre_block_seg_end_coordinate < 10
+                    )
+                    prev_lines_more_than_one = num_of_prev_lines > 1
+
+                    overlap_blocks = context_left_coordinate < prev_block_bbox[2]
 
                     # update context_left_coordinate and context_right_coordinate
-                    if context_left_coordinate < prev_block_bbox[2]:
+                    if overlap_blocks:
                         context_left_coordinate = min(
                             prev_block_bbox[0], context_left_coordinate
                         )
                         context_right_coordinate = max(
                             prev_block_bbox[2], context_right_coordinate
                         )
+                        prev_end_space_small = (
+                            prev_block_bbox[2] - pre_block_seg_end_coordinate < 10
+                        )
 
-                    # 判断是否需要分段
-                    prev_end_space_small = (
-                        prev_block_bbox[2] - pre_block_seg_end_coordinate < 10
-                    )
                     current_start_space_small = (
                         seg_start_coordinate - context_left_coordinate < 10
                     )
-                    overlap_blocks = context_left_coordinate < prev_block_bbox[2]
-                    prev_lines_more_than_one = num_of_prev_lines > 1
 
                     if (
-                        overlap_blocks
-                        and current_start_space_small
-                        and prev_lines_more_than_one
-                    ) or (
                         prev_end_space_small
                         and current_start_space_small
                         and prev_lines_more_than_one