소스 검색

fix build index when init

gaotingquan 8 달 전
부모
커밋
59b0cae2fb
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      paddlex/inference/pipelines/pp_shitu_v2/pipeline.py

+ 1 - 1
paddlex/inference/pipelines/pp_shitu_v2/pipeline.py

@@ -50,7 +50,7 @@ class ShiTuV2Pipeline(BasePipeline):
         self.det_model = self.create_model(config["SubModules"]["Detection"])
         self.rec_model = self.create_model(config["SubModules"]["Recognition"])
         self.crop_by_boxes = CropByBoxes()
-        self.indexer = self._build_indexer(index=index) if index else None
+        self.indexer = FaissIndexer(index=index) if index else None
         self.batch_sampler = ImageBatchSampler(
             batch_size=self.det_model.batch_sampler.batch_size
         )