Skip to content

P
pdf-miner
Commit 278ba2f6 authored by myhloli's avatar myhloli
Browse files
Options

update error output

parent 415a0105
Show whitespace changes
Inline Side-by-side
Showing with 24 additions and 20 deletions
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 loguru import logger
import os
from paddleocr import draw_ocr try:
from PIL import Image import cv2
from torchvision import transforms import yaml
from torch.utils.data import Dataset, DataLoader import time
from ultralytics import YOLO import argparse
from unimernet.common.config import Config import numpy as np
import unimernet.tasks as tasks import torch
from unimernet.processors import load_processor
from paddleocr import draw_ocr
from magic_pdf.model.pek_sub_modules.layoutlmv3.model_init import Layoutlmv3_Predictor from PIL import Image
from magic_pdf.model.pek_sub_modules.post_process import get_croped_image, latex_rm_whitespace from torchvision import transforms
from magic_pdf.model.pek_sub_modules.self_modify import ModifiedPaddleOCR 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): def mfd_model_init(weight):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment