Commit 4bf58088 authored by myhloli's avatar myhloli

fix(magic-pdf): prevent Albumentations update check

parent 0d9427ca
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')
# 裁剪图像 # 裁剪图像
......
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