mirror of
https://github.com/opendatalab/MinerU.git
synced 2026-09-24 23:10:23 +08:00
@@ -282,8 +282,6 @@ class CustomPEKModel:
|
||||
)
|
||||
# 初始化ocr
|
||||
if self.apply_ocr:
|
||||
|
||||
# self.ocr_model = ModifiedPaddleOCR(show_log=show_log, det_db_box_thresh=0.3)
|
||||
self.ocr_model = atom_model_manager.get_atom_model(
|
||||
atom_model_name=AtomicModel.OCR,
|
||||
ocr_show_log=show_log,
|
||||
@@ -301,17 +299,6 @@ class CustomPEKModel:
|
||||
device=self.device
|
||||
)
|
||||
|
||||
home_directory = Path.home()
|
||||
det_source = os.path.join(models_dir, table_model_dir, DETECT_MODEL_DIR)
|
||||
rec_source = os.path.join(models_dir, table_model_dir, REC_MODEL_DIR)
|
||||
det_dest_dir = os.path.join(home_directory, PP_DET_DIRECTORY)
|
||||
rec_dest_dir = os.path.join(home_directory, PP_REC_DIRECTORY)
|
||||
|
||||
if not os.path.exists(det_dest_dir):
|
||||
shutil.copytree(det_source, det_dest_dir)
|
||||
if not os.path.exists(rec_dest_dir):
|
||||
shutil.copytree(rec_source, rec_dest_dir)
|
||||
|
||||
logger.info('DocAnalysis init done!')
|
||||
|
||||
def __call__(self, image):
|
||||
|
||||
Reference in New Issue
Block a user