2020-06-16 00:05:18 +08:00
|
|
|
from .builder import build_dataset
|
2020-06-29 00:10:34 +08:00
|
|
|
from .byol import BYOLDataset
|
2020-06-16 00:05:18 +08:00
|
|
|
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
|
2020-07-08 22:30:48 +08:00
|
|
|
from .relative_loc import RelativeLocDataset
|
2020-06-16 00:05:18 +08:00
|
|
|
from .contrastive import ContrastiveDataset
|
|
|
|
|
from .dataset_wrappers import ConcatDataset, RepeatDataset
|
|
|
|
|
from .loader import DistributedGroupSampler, GroupSampler, build_dataloader
|
|
|
|
|
from .registry import DATASETS
|
