42 lines
2.0 KiB
Python
42 lines
2.0 KiB
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .ade import ADE20KDataset
|
|
from .basesegdataset import BaseSegDataset
|
|
from .chase_db1 import ChaseDB1Dataset
|
|
from .cityscapes import CityscapesDataset
|
|
from .coco_stuff import COCOStuffDataset
|
|
from .dark_zurich import DarkZurichDataset
|
|
from .dataset_wrappers import MultiImageMixDataset
|
|
from .decathlon import DecathlonDataset
|
|
from .drive import DRIVEDataset
|
|
from .hrf import HRFDataset
|
|
from .isaid import iSAIDDataset
|
|
from .isprs import ISPRSDataset
|
|
from .lip import LIPDataset
|
|
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, GenerateEdge, LoadAnnotations,
|
|
LoadBiomedicalAnnotation, LoadBiomedicalData,
|
|
LoadBiomedicalImageFromFile, LoadImageFromNDArray,
|
|
PackSegInputs, PhotoMetricDistortion, RandomCrop,
|
|
RandomCutOut, RandomMosaic, RandomRotate, Rerange,
|
|
ResizeShortestEdge, 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', 'LoadBiomedicalImageFromFile',
|
|
'LoadBiomedicalAnnotation', 'LoadBiomedicalData', 'GenerateEdge',
|
|
'DecathlonDataset', 'LIPDataset', 'ResizeShortestEdge'
|
|
]
|