from .registry import DATASETS from .base import BaseDataset @DATASETS.register_module class ExtractDataset(BaseDataset): """Dataset for feature extraction. """ def __init__(self, data_source, pipeline): super(ExtractDataset, self).__init__(data_source, pipeline) def __getitem__(self, idx): img = self.data_source.get_sample(idx) img = self.pipeline(img) return dict(img=img) def evaluate(self, scores, keyword, logger=None): raise NotImplemented