فهرست منبع

Fix SLANet_plus infer bug (#2871)

* add table models

* refine codes

* refine codes

* fix bugs
Liu Jiaxuan 10 ماه پیش
والد
کامیت
47cda8aaac
1فایلهای تغییر یافته به همراه1 افزوده شده و 1 حذف شده
  1. 1 1
      paddlex/inference/models_new/table_structure_recognition/predictor.py

+ 1 - 1
paddlex/inference/models_new/table_structure_recognition/predictor.py

@@ -66,7 +66,7 @@ class TablePredictor(BasicPredictor):
         )
 
         postprocessors = TableLabelDecode(
-            model_name="SLANet",
+            model_name=self.config["Global"]["model_name"],
             merge_no_span_structure=self.config["PreProcess"]["transform_ops"][1][
                 "TableLabelEncode"
             ]["merge_no_span_structure"],