EasyCV/easycv/datasets/shared/raw.py

# 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