mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[Fix] Fix pipeline bug in image retrieval inferencer

pull/1815/head
mzr1996 2023-10-08 15:44:37 +08:00
parent bb59c9ad82
commit 06bb586eb7
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mmpretrain/apis/image_retrieval.py
View File

@ -108,6 +108,7 @@ class ImageRetrievalInferencer(BaseInferencer):
# A config of dataset # A config of dataset
from mmpretrain.registry import DATASETS from mmpretrain.registry import DATASETS
test_pipeline = [dict(type='LoadImageFromFile'), self.pipeline] test_pipeline = [dict(type='LoadImageFromFile'), self.pipeline]
prototype.setdefault('pipeline', test_pipeline)
dataset = DATASETS.build(prototype) dataset = DATASETS.build(prototype)
dataloader = build_dataloader(dataset) dataloader = build_dataloader(dataset)
elif isinstance(prototype, DataLoader): elif isinstance(prototype, DataLoader):