Ver código fonte

change layout model for table rec (#2894)

* Update paddlepaddle_install.md

* Update paddlepaddle_install_en.md

* add table pipev2

* Update test_table_recognition_v2.py

* Update test_table_recognition_v2.py

* change infer models

* Update table_recognition.yaml

* Update table_recognition_v2.yaml
Liu Jiaxuan 10 meses atrás
pai
commit
06271c4a10

+ 1 - 1
paddlex/configs/pipelines/table_recognition.yaml

@@ -8,7 +8,7 @@ use_ocr_model: True
 SubModules:
   LayoutDetection:
     module_name: layout_detection
-    model_name: RT-DETR-H_layout_17cls
+    model_name: PicoDet_layout_1x_table
     model_dir: null
 
   TableStructureRecognition:

+ 1 - 1
paddlex/configs/pipelines/table_recognition_v2.yaml

@@ -8,7 +8,7 @@ use_ocr_model: True
 SubModules:
   LayoutDetection:
     module_name: layout_detection
-    model_name: RT-DETR-H_layout_17cls
+    model_name: PicoDet_layout_1x_table
     model_dir: null
   
   TableClassification: