소스 검색

Merge pull request #85 from papayalove/master

解决list未换行问题
myhloli 1 년 전
부모
커밋
c8ccc39078
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      magic_pdf/para/para_split_v2.py

+ 1 - 1
magic_pdf/para/para_split_v2.py

@@ -40,7 +40,7 @@ def __detect_list_lines(lines, new_layout_bboxes, lang):
                     if lst[i] == 1:
                         ones_in_this_interval.append(i)
                     i += 1
-                if len(ones_in_this_interval) > 1 or (ones_in_this_interval and lst[start + 1] in [2, 3]):
+                if len(ones_in_this_interval) > 1 or (start < len(lst) - 1 and ones_in_this_interval and lst[start + 1] in [2, 3]):
                     indices.append((start, i - 1))
                     ones_indices.append(ones_in_this_interval)
             else: