|
|
@@ -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):
|