EasyCV/easycv/datasets/__init__.py

13 lines
634 B
Python

# Copyright (c) Alibaba, Inc. and its affiliates.
# isort:skip_file
from easycv.utils.import_utils import check_numpy
check_numpy()
from . import (classification, detection, detection3d, face, ocr, pose,
segmentation, selfsup, shared)
from .builder import (build_dali_dataset, build_dataset, build_datasource,
build_sampler)
from .loader import (DistributedGivenIterationSampler, DistributedGroupSampler,
DistributedMPSampler, DistributedSampler, GroupSampler,
RASampler, build_dataloader)
from .registry import DATASETS, DATASOURCES, PIPELINES, SAMPLERS