fix build index when init

TingquanGao · TingquanGao · commit 3de9604c9dec · 2025-03-05T19:47:01.000+08:00
diff --git a/paddlex/inference/pipelines/pp_shitu_v2/pipeline.py b/paddlex/inference/pipelines/pp_shitu_v2/pipeline.py
@@ -50,7 +50,7 @@ def __init__(
         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
         )

Original file line number	Diff line number	Diff line change
`@@ -50,7 +50,7 @@ def __init__(`
`50`	`50`	`self.det_model = self.create_model(config["SubModules"]["Detection"])`
`51`	`51`	`self.rec_model = self.create_model(config["SubModules"]["Recognition"])`
`52`	`52`	`self.crop_by_boxes = CropByBoxes()`
`53`		`- self.indexer = self._build_indexer(index=index) if index else None`
	`53`	`+ self.indexer = FaissIndexer(index=index) if index else None`
`54`	`54`	`self.batch_sampler = ImageBatchSampler(`
`55`	`55`	`batch_size=self.det_model.batch_sampler.batch_size`
`56`	`56`	`)`