13 lines
508 B
Python
Raw Normal View History

2020-06-16 00:05:18 +08:00
from .builder import build_dataset
from .data_sources import *
from .pipelines import *
from .classification import ClassificationDataset
from .deepcluster import DeepClusterDataset
from .extraction import ExtractDataset
from .npid import NPIDDataset
from .rotation_pred import RotationPredDataset
from .contrastive import ContrastiveDataset
from .dataset_wrappers import ConcatDataset, RepeatDataset
from .loader import DistributedGroupSampler, GroupSampler, build_dataloader
from .registry import DATASETS