36 lines
1.6 KiB
Python
36 lines
1.6 KiB
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .ade import ADE20KDataset
|
|
from .chase_db1 import ChaseDB1Dataset
|
|
from .cityscapes import CityscapesDataset
|
|
from .coco_stuff import COCOStuffDataset
|
|
from .custom import BaseSegDataset
|
|
from .dark_zurich import DarkZurichDataset
|
|
from .dataset_wrappers import MultiImageMixDataset
|
|
from .drive import DRIVEDataset
|
|
from .hrf import HRFDataset
|
|
from .isaid import iSAIDDataset
|
|
from .isprs import ISPRSDataset
|
|
from .loveda import LoveDADataset
|
|
from .night_driving import NightDrivingDataset
|
|
from .pascal_context import PascalContextDataset, PascalContextDataset59
|
|
from .potsdam import PotsdamDataset
|
|
from .stare import STAREDataset
|
|
from .transforms import (CLAHE, AdjustGamma, LoadAnnotations,
|
|
LoadImageFromNDArray, PackSegInputs,
|
|
PhotoMetricDistortion, RandomCrop, RandomCutOut,
|
|
RandomMosaic, RandomRotate, Rerange, ResizeToMultiple,
|
|
RGB2Gray, SegRescale)
|
|
from .voc import PascalVOCDataset
|
|
|
|
__all__ = [
|
|
'BaseSegDataset', 'CityscapesDataset', 'PascalVOCDataset', 'ADE20KDataset',
|
|
'PascalContextDataset', 'PascalContextDataset59', 'ChaseDB1Dataset',
|
|
'DRIVEDataset', 'HRFDataset', 'STAREDataset', 'DarkZurichDataset',
|
|
'NightDrivingDataset', 'COCOStuffDataset', 'LoveDADataset',
|
|
'MultiImageMixDataset', 'iSAIDDataset', 'ISPRSDataset', 'PotsdamDataset',
|
|
'LoadAnnotations', 'RandomCrop', 'SegRescale', 'PhotoMetricDistortion',
|
|
'RandomRotate', 'AdjustGamma', 'CLAHE', 'Rerange', 'RGB2Gray',
|
|
'RandomCutOut', 'RandomMosaic', 'PackSegInputs', 'ResizeToMultiple',
|
|
'LoadImageFromNDArray'
|
|
]
|