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
278ba2f6
Commit
278ba2f6
authored
Jul 12, 2024
by
myhloli
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
update error output
parent
415a0105
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
24 additions
and
20 deletions
+24
-20
pdf_extract_kit.py
magic_pdf/model/pdf_extract_kit.py
+24
-20
No files found.
magic_pdf/model/pdf_extract_kit.py
View file @
278ba2f6
import
os
import
cv2
import
yaml
import
time
import
argparse
import
numpy
as
np
import
torch
from
loguru
import
logger
from
paddleocr
import
draw_ocr
from
PIL
import
Image
from
torchvision
import
transforms
from
torch.utils.data
import
Dataset
,
DataLoader
from
ultralytics
import
YOLO
from
unimernet.common.config
import
Config
import
unimernet.tasks
as
tasks
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
import
os
try
:
import
cv2
import
yaml
import
time
import
argparse
import
numpy
as
np
import
torch
from
paddleocr
import
draw_ocr
from
PIL
import
Image
from
torchvision
import
transforms
from
torch.utils.data
import
Dataset
,
DataLoader
from
ultralytics
import
YOLO
from
unimernet.common.config
import
Config
import
unimernet.tasks
as
tasks
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
:
logger
.
error
(
'Required dependency not installed, please install by
\n
"pip install magic-pdf[full-cpu] detectron2 --extra-index-url https://myhloli.github.io/wheels/"'
)
exit
(
1
)
def
mfd_model_init
(
weight
):
...
...
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