Просмотр исходного кода

update model name & don't use oricls_model and uvdoc_model by default

gaotingquan 1 год назад
Родитель
Сommit
c22b998614

+ 3 - 3
paddlex/inference/pipelines/ppchatocrv3/ppchatocrv3.py

@@ -43,9 +43,9 @@ class PPChatOCRPipeline(TableRecPipeline):
         text_det_model,
         text_rec_model,
         table_model,
-        oricls_model=None,
-        uvdoc_model=None,
-        curve_model=None,
+        doc_image_ori_cls_model=None,
+        doc_image_unwarp_model=None,
+        seal_text_det_model=None,
         llm_name="ernie-3.5",
         llm_params={},
         task_prompt_yaml=None,

+ 3 - 3
paddlex/pipelines/PP-ChatOCRv3-doc.yaml

@@ -7,9 +7,9 @@ Pipeline:
   table_model: SLANet_plus
   text_det_model: PP-OCRv4_server_det
   text_rec_model: PP-OCRv4_server_rec
-  uvdoc_model: UVDoc
-  curve_model: PP-OCRv4_server_seal_det
-  oricls_model: PP-LCNet_x1_0_doc_ori
+  seal_text_det_model: PP-OCRv4_server_seal_det
+  doc_image_ori_cls_model: null # PP-LCNet_x1_0_doc_ori
+  doc_image_unwarp_model: null # UVDoc
   llm_name: "ernie-3.5"
   llm_params: 
     api_type: qianfan