mirror of
https://github.com/deepinsight/insightface.git
synced 2026-05-14 12:17:55 +00:00
25 lines
1.1 KiB
Python
Executable File
25 lines
1.1 KiB
Python
Executable File
from .builder import DATASETS, PIPELINES, build_dataloader, build_dataset
|
|
from .cityscapes import CityscapesDataset
|
|
from .coco import CocoDataset
|
|
from .custom import CustomDataset
|
|
from .retinaface import RetinaFaceDataset
|
|
from .dataset_wrappers import (ClassBalancedDataset, ConcatDataset,
|
|
RepeatDataset)
|
|
from .deepfashion import DeepFashionDataset
|
|
from .lvis import LVISDataset, LVISV1Dataset, LVISV05Dataset
|
|
from .samplers import DistributedGroupSampler, DistributedSampler, GroupSampler
|
|
from .utils import replace_ImageToTensor
|
|
from .voc import VOCDataset
|
|
from .wider_face import WIDERFaceDataset
|
|
from .xml_style import XMLDataset
|
|
|
|
__all__ = [
|
|
'CustomDataset', 'XMLDataset', 'CocoDataset', 'DeepFashionDataset',
|
|
'VOCDataset', 'CityscapesDataset', 'LVISDataset', 'LVISV05Dataset',
|
|
'LVISV1Dataset', 'GroupSampler', 'DistributedGroupSampler',
|
|
'DistributedSampler', 'build_dataloader', 'ConcatDataset', 'RepeatDataset',
|
|
'RetinaFaceDataset',
|
|
'ClassBalancedDataset', 'WIDERFaceDataset', 'DATASETS', 'PIPELINES',
|
|
'build_dataset', 'replace_ImageToTensor'
|
|
]
|