mirror of https://github.com/alibaba/EasyCV.git
18 lines
508 B
Python
18 lines
508 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.get_sample(idx)
|
|
return self.pipeline(results)
|
|
|
|
def evaluate(self, scores, keyword, logger=None):
|
|
raise NotImplementedError
|