Переглянути джерело

feat: merge formula update

liukaiwen 1 рік тому
батько
коміт
51f56aa32f
2 змінених файлів з 6 додано та 2 видалено
  1. 2 2
      magic_pdf/libs/Constants.py
  2. 4 0
      magic_pdf/model/pdf_extract_kit.py

+ 2 - 2
magic_pdf/libs/Constants.py

@@ -29,10 +29,10 @@ TABLE_MASTER_DICT = "table_master_structure_dict.txt"
 TABLE_MASTER_DIR = "table_structure_tablemaster_infer/"
 
 # pp detect model dir
-DETECT_MODEL_DIR = "ch_PP-OCRv3_det_infer"
+DETECT_MODEL_DIR = "ch_PP-OCRv4_det_infer"
 
 # pp rec model dir
-REC_MODEL_DIR = "ch_PP-OCRv3_rec_infer"
+REC_MODEL_DIR = "ch_PP-OCRv4_rec_infer"
 
 # pp rec char dict path
 REC_CHAR_DICT = "ppocr_keys_v1.txt"

+ 4 - 0
magic_pdf/model/pdf_extract_kit.py

@@ -433,3 +433,7 @@ class CustomPEKModel:
         logger.info(f"-----page total time: {round(time.time() - page_start, 2)}-----")
 
         return layout_res
+if __name__ == '__main__':
+    print()
+
+