Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
wangsen
MinerU
Commits
21cfaf4c
Commit
21cfaf4c
authored
Mar 13, 2024
by
liukaiwen
Browse files
add modify inline equation y axis
add false displayed equation to inline equation
parent
64d67b5c
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
9 additions
and
4 deletions
+9
-4
magic_pdf/pre_proc/ocr_span_list_modify.py
magic_pdf/pre_proc/ocr_span_list_modify.py
+9
-4
No files found.
magic_pdf/pre_proc/ocr_span_list_modify.py
View file @
21cfaf4c
from
magic_pdf.libs.boxbase
import
calculate_overlap_area_in_bbox1_area_ratio
,
get_minbox_if_overlap_by_ratio
from
magic_pdf.libs.boxbase
import
__is_overlaps_y_exceeds_threshold
def
remove_overlaps_min_spans
(
spans
):
# 删除重叠spans中较小的那些
...
...
@@ -127,16 +127,21 @@ def modify_inline_equation(spans: list, displayed_list: list, text_inline_lines:
text_line
=
text_inline_lines
[
j
]
y0
,
y1
=
text_line
[
1
]
if
(
span_y0
<
y0
and
span_y
>
y0
or
span_y0
<
y1
and
span_y
>
y1
or
span_y0
<
y0
and
span_y
>
y1
)
and
__is_overlaps_y_exceeds_threshold
(
span
[
'bbox'
],
(
0
,
y0
,
0
,
y1
)):
span
[
"bbox"
][
1
]
=
y0
# span["bbox"][3] = y1
#调整公式类型
if
span
[
"type"
]
==
"displayed_equation"
:
#最后一行是行间公式
if
j
+
1
>=
len
(
text_inline_lines
):
span
[
"type"
]
=
"inline_equation"
span
[
"bbox"
][
1
]
=
y0
span
[
"bbox"
][
3
]
=
y1
else
:
#行间公式旁边有多行文字或者行间公式比文字高3倍则不转换
y0_next
,
y1_next
=
text_inline_lines
[
j
+
1
][
1
]
if
not
__is_overlaps_y_exceeds_threshold
(
span
[
'bbox'
],
(
0
,
y0_next
,
0
,
y1_next
)):
if
not
__is_overlaps_y_exceeds_threshold
(
span
[
'bbox'
],
(
0
,
y0_next
,
0
,
y1_next
))
and
3
*
(
y1
-
y0
)
>
span_y
-
span_y0
:
span
[
"type"
]
=
"inline_equation"
span
[
"bbox"
][
1
]
=
y0
span
[
"bbox"
][
3
]
=
y1
break
elif
span_y
<
y0
or
span_y0
<
y0
and
span_y
>
y0
and
not
__is_overlaps_y_exceeds_threshold
(
span
[
'bbox'
],
(
0
,
y0
,
0
,
y1
)):
break
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment