Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
P
pdf-miner
Project
Project
Details
Activity
Releases
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Qin Kaijie
pdf-miner
Commits
4bf58088
Commit
4bf58088
authored
Jul 24, 2024
by
myhloli
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
fix(magic-pdf): prevent Albumentations update check
parent
0d9427ca
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
11 additions
and
6 deletions
+11
-6
pdf_extract_kit.py
magic_pdf/model/pdf_extract_kit.py
+11
-6
No files found.
magic_pdf/model/pdf_extract_kit.py
View file @
4bf58088
from
loguru
import
logger
from
loguru
import
logger
import
os
import
os
import
time
import
time
os
.
environ
[
'NO_ALBUMENTATIONS_UPDATE'
]
=
'1'
# 禁止albumentations检查更新
try
:
try
:
import
cv2
import
cv2
import
yaml
import
yaml
...
@@ -17,14 +19,17 @@ try:
...
@@ -17,14 +19,17 @@ try:
import
unimernet.tasks
as
tasks
import
unimernet.tasks
as
tasks
from
unimernet.processors
import
load_processor
from
unimernet.processors
import
load_processor
from
magic_pdf.model.pek_sub_modules.layoutlmv3.model_init
import
Layoutlmv3_Predictor
from
magic_pdf.model.pek_sub_modules.post_process
import
get_croped_image
,
latex_rm_whitespace
from
magic_pdf.model.pek_sub_modules.self_modify
import
ModifiedPaddleOCR
except
ImportError
as
e
:
except
ImportError
as
e
:
logger
.
exception
(
e
)
logger
.
exception
(
e
)
logger
.
error
(
'Required dependency not installed, please install by
\n
"pip install magic-pdf[full] detectron2 --extra-index-url https://myhloli.github.io/wheels/"'
)
logger
.
error
(
'Required dependency not installed, please install by
\n
'
'"pip install magic-pdf[full] detectron2 --extra-index-url https://myhloli.github.io/wheels/"'
)
exit
(
1
)
exit
(
1
)
from
magic_pdf.model.pek_sub_modules.layoutlmv3.model_init
import
Layoutlmv3_Predictor
from
magic_pdf.model.pek_sub_modules.post_process
import
get_croped_image
,
latex_rm_whitespace
from
magic_pdf.model.pek_sub_modules.self_modify
import
ModifiedPaddleOCR
def
mfd_model_init
(
weight
):
def
mfd_model_init
(
weight
):
mfd_model
=
YOLO
(
weight
)
mfd_model
=
YOLO
(
weight
)
...
@@ -189,8 +194,8 @@ class CustomPEKModel:
...
@@ -189,8 +194,8 @@ class CustomPEKModel:
paste_x
=
50
paste_x
=
50
paste_y
=
50
paste_y
=
50
# 创建一个宽高各多50的白色背景
# 创建一个宽高各多50的白色背景
new_width
=
xmax
-
xmin
+
paste_x
*
2
new_width
=
xmax
-
xmin
+
paste_x
*
2
new_height
=
ymax
-
ymin
+
paste_y
*
2
new_height
=
ymax
-
ymin
+
paste_y
*
2
new_image
=
Image
.
new
(
'RGB'
,
(
new_width
,
new_height
),
'white'
)
new_image
=
Image
.
new
(
'RGB'
,
(
new_width
,
new_height
),
'white'
)
# 裁剪图像
# 裁剪图像
...
...
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