[Fix] Fix pipeline bug in image retrieval inferencer

2023-10-08 15:44:37 +08:00 · 2023-10-08 15:44:37 +08:00 · 06bb586eb7
parent bb59c9ad82
commit 06bb586eb7
1 changed files with 1 additions and 0 deletions
--- a/mmpretrain/apis/image_retrieval.py
+++ b/mmpretrain/apis/image_retrieval.py
@ -108,6 +108,7 @@ class ImageRetrievalInferencer(BaseInferencer):
            # A config of dataset
            from mmpretrain.registry import DATASETS
            test_pipeline = [dict(type='LoadImageFromFile'), self.pipeline]
+            prototype.setdefault('pipeline', test_pipeline)
            dataset = DATASETS.build(prototype)
            dataloader = build_dataloader(dataset)
        elif isinstance(prototype, DataLoader):