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_det_batch_size = text_det_batch_size
         self.text_rec_batch_size = text_rec_batch_size
         self.text_rec_batch_size = text_rec_batch_size
         self.table_batch_size = table_batch_size
         self.table_batch_size = table_batch_size
+        self.predictor_kwargs = predictor_kwargs
         super().__init__(predictor_kwargs=predictor_kwargs)
         super().__init__(predictor_kwargs=predictor_kwargs)
         self._build_predictor()
         self._build_predictor()
         # self.set_predictor(layout_batch_size, text_det_batch_size,text_rec_batch_size, table_batch_size)
         # self.set_predictor(layout_batch_size, text_det_batch_size,text_rec_batch_size, table_batch_size)