mirror of
https://github.com/NVlabs/SegFormer.git
synced 2025-06-03 14:48:49 +08:00
21 lines
850 B
Python
Executable File
21 lines
850 B
Python
Executable File
from .ade import ADE20KDataset
|
|
from .builder import DATASETS, PIPELINES, build_dataloader, build_dataset
|
|
from .chase_db1 import ChaseDB1Dataset
|
|
from .cityscapes import CityscapesDataset
|
|
from .custom import CustomDataset
|
|
from .dataset_wrappers import ConcatDataset, RepeatDataset
|
|
from .drive import DRIVEDataset
|
|
from .hrf import HRFDataset
|
|
from .pascal_context import PascalContextDataset
|
|
from .stare import STAREDataset
|
|
from .voc import PascalVOCDataset
|
|
from .mapillary import MapillaryDataset
|
|
from .cocostuff import CocoStuff
|
|
|
|
__all__ = [
|
|
'CustomDataset', 'build_dataloader', 'ConcatDataset', 'RepeatDataset',
|
|
'DATASETS', 'build_dataset', 'PIPELINES', 'CityscapesDataset',
|
|
'PascalVOCDataset', 'ADE20KDataset', 'PascalContextDataset',
|
|
'ChaseDB1Dataset', 'DRIVEDataset', 'HRFDataset', 'STAREDataset', 'MapillaryDataset', 'CocoStuff'
|
|
]
|