소스 검색

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"],