[Fix] Fix pipeline bug in image retrieval inferencer

open-mmlab · Oct 8, 2023 · 06bb586 · 06bb586
1 parent bb59c9a
commit 06bb586
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mmpretrain/apis/image_retrieval.py b/mmpretrain/apis/image_retrieval.py
@@ -108,6 +108,7 @@ def build_dataloader(dataset):
             # 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):