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

Merge pull request #1835 from changdazhou/fix_bug_dev

fix predict bug
cuicheng01 1 год назад
Родитель
Сommit
3467290b12

+ 1 - 1
paddlex/configs/structure_analysis/PicoDet_layout_1x.yaml

@@ -30,7 +30,7 @@ Evaluate:
   log_interval: 10
 
 Predict:
-  model: "output/best_model"
+  model_dir: "output/best_model"
   input_path: "/paddle/dataset/paddlex/layout/det_layout_examples/images/JPEGImages/train_0001.jpg"
   kernel_option:
     run_mode: paddle

+ 1 - 3
paddlex/modules/table_recognition/predictor/predictor.py

@@ -12,8 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-
-
 import os
 
 import numpy as np
@@ -38,6 +36,7 @@ class TableRecPredictor(BasePredictor):
                  pre_transforms=None,
                  post_transforms=None,
                  table_max_len=488):
+        self.table_max_len = table_max_len
         super().__init__(
             model_name=model_name,
             model_dir=model_dir,
@@ -45,7 +44,6 @@ class TableRecPredictor(BasePredictor):
             output=output,
             pre_transforms=pre_transforms,
             post_transforms=post_transforms)
-        self.table_max_len = table_max_len
 
     @classmethod
     def get_input_keys(cls):