EasyCV/easycv/datasets/__init__.py

13 lines
634 B
Python
Raw Normal View History

2022-04-02 20:01:06 +08:00
# 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