浏览代码

解决list未换行问题

liukaiwen 1 年之前
父节点
当前提交
69d835c998
共有 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:
                     if lst[i] == 1:
                         ones_in_this_interval.append(i)
                         ones_in_this_interval.append(i)
                     i += 1
                     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))
                     indices.append((start, i - 1))
                     ones_indices.append(ones_in_this_interval)
                     ones_indices.append(ones_in_this_interval)
             else:
             else: