Cathy0908 a11f200ec3
refactor get_sample to __getitem__ for all datasources api (#156)
* refactor `get_sample` to `__getitem__` for all datasources api

* fix MMpad config
2022-08-17 14:24:17 +08:00

18 lines
497 B
Python

# Copyright (c) Alibaba, Inc. and its affiliates.
from easycv.datasets.registry import DATASETS
from .base import BaseDataset
@DATASETS.register_module
class RawDataset(BaseDataset):
def __init__(self, data_source, pipeline):
super(RawDataset, self).__init__(data_source, pipeline)
def __getitem__(self, idx):
results = self.data_source[idx]
return self.pipeline(results)
def evaluate(self, scores, keyword, logger=None):
raise NotImplementedError