gaotingquan преди 1 година
родител
ревизия
8a4726d7b6
променени са 1 файла, в които са добавени 1 реда и са изтрити 0 реда
  1. 1 0
      paddlex/inference/pipelines/table_recognition/table_recognition.py

+ 1 - 0
paddlex/inference/pipelines/table_recognition/table_recognition.py

@@ -46,6 +46,7 @@ class TableRecPipeline(BasePipeline):
         self.text_det_batch_size = text_det_batch_size
         self.text_rec_batch_size = text_rec_batch_size
         self.table_batch_size = table_batch_size
+        self.predictor_kwargs = predictor_kwargs
         super().__init__(predictor_kwargs=predictor_kwargs)
         self._build_predictor()
         # self.set_predictor(layout_batch_size, text_det_batch_size,text_rec_batch_size, table_batch_size)