|
@@ -1,5 +1,3 @@
|
|
|
-import concurrent.futures as fut
|
|
|
|
|
-import multiprocessing as mp
|
|
|
|
|
import os
|
|
import os
|
|
|
import time
|
|
import time
|
|
|
|
|
|
|
@@ -25,8 +23,6 @@ from magic_pdf.libs.config_reader import (get_device, get_formula_config,
|
|
|
get_table_recog_config)
|
|
get_table_recog_config)
|
|
|
from magic_pdf.model.model_list import MODEL
|
|
from magic_pdf.model.model_list import MODEL
|
|
|
|
|
|
|
|
-# from magic_pdf.operators.models import InferenceResult
|
|
|
|
|
-
|
|
|
|
|
class ModelSingleton:
|
|
class ModelSingleton:
|
|
|
_instance = None
|
|
_instance = None
|
|
|
_models = {}
|
|
_models = {}
|
|
@@ -244,9 +240,7 @@ def may_batch_image_analyze(
|
|
|
formula_enable=None,
|
|
formula_enable=None,
|
|
|
table_enable=None):
|
|
table_enable=None):
|
|
|
# os.environ['CUDA_VISIBLE_DEVICES'] = str(idx)
|
|
# os.environ['CUDA_VISIBLE_DEVICES'] = str(idx)
|
|
|
- # 关闭paddle的信号处理
|
|
|
|
|
- import paddle
|
|
|
|
|
- paddle.disable_signal_handler()
|
|
|
|
|
|
|
+
|
|
|
from magic_pdf.model.batch_analyze import BatchAnalyze
|
|
from magic_pdf.model.batch_analyze import BatchAnalyze
|
|
|
|
|
|
|
|
model_manager = ModelSingleton()
|
|
model_manager = ModelSingleton()
|