Răsfoiți Sursa

update PP-ChatOCRv4-doc yaml (#3322)

changdazhou 9 luni în urmă
părinte
comite
4a1a1eed28
1 a modificat fișierele cu 8 adăugiri și 33 ștergeri
  1. 8 33
      paddlex/configs/pipelines/PP-ChatOCRv4-doc.yaml

+ 8 - 33
paddlex/configs/pipelines/PP-ChatOCRv4-doc.yaml

@@ -93,7 +93,7 @@ SubModules:
 
 SubPipelines:
   LayoutParser:
-    pipeline_name: layout_parsing_v2
+    pipeline_name: layout_parsing
 
     use_doc_preprocessor: True
     use_general_ocr: True
@@ -104,13 +104,8 @@ SubPipelines:
     SubModules:
       LayoutDetection:
         module_name: layout_detection
-        model_name: PP-DocLayout-L
+        model_name: RT-DETR-H_layout_3cls
         model_dir: null
-        threshold: 
-          7: 0.3
-        layout_nms: True
-        layout_unclip_ratio: 1.0
-        layout_merge_bboxes_mode: "large"
 
     SubPipelines:
       DocPreprocessor:
@@ -147,38 +142,18 @@ SubPipelines:
             module_name: text_recognition
             model_name: PP-OCRv4_server_rec_doc
             model_dir: null
-            batch_size: 1
-            score_thresh: 0.0
+            batch_size: 6
+            score_thresh: 0
 
       TableRecognition:
-        pipeline_name: table_recognition_v2
+        pipeline_name: table_recognition
         use_layout_detection: False
         use_doc_preprocessor: False
         use_ocr_model: False
-        SubModules:  
-          TableClassification:
-            module_name: table_classification
-            model_name: PP-LCNet_x1_0_table_cls
-            model_dir: null
-
-          WiredTableStructureRecognition:
-            module_name: table_structure_recognition
-            model_name: SLANeXt_wired
-            model_dir: null
-          
-          WirelessTableStructureRecognition:
+        SubModules:
+          TableStructureRecognition:
             module_name: table_structure_recognition
-            model_name: SLANeXt_wireless
-            model_dir: null
-          
-          WiredTableCellsDetection:
-            module_name: table_cells_detection
-            model_name: RT-DETR-L_wired_table_cell_det
-            model_dir: null
-          
-          WirelessTableCellsDetection:
-            module_name: table_cells_detection
-            model_name: RT-DETR-L_wireless_table_cell_det
+            model_name: SLANet_plus
             model_dir: null
 
       SealRecognition: