2020-07-08 22:30:48 +08:00

15 lines
583 B
Python

from .builder import build_dataset
from .byol import BYOLDataset
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 .relative_loc import RelativeLocDataset
from .contrastive import ContrastiveDataset
from .dataset_wrappers import ConcatDataset, RepeatDataset
from .loader import DistributedGroupSampler, GroupSampler, build_dataloader
from .registry import DATASETS