61 lines
3.0 KiB
Python
61 lines
3.0 KiB
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
# yapf: disable
|
|
from .ade import ADE20KDataset
|
|
from .basesegdataset import BaseCDDataset, 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 .levir import LEVIRCDDataset
|
|
from .lip import LIPDataset
|
|
from .loveda import LoveDADataset
|
|
from .mapillary import MapillaryDataset_v1, MapillaryDataset_v2
|
|
from .night_driving import NightDrivingDataset
|
|
from .pascal_context import PascalContextDataset, PascalContextDataset59
|
|
from .potsdam import PotsdamDataset
|
|
from .refuge import REFUGEDataset
|
|
from .stare import STAREDataset
|
|
from .synapse import SynapseDataset
|
|
# yapf: disable
|
|
from .transforms import (CLAHE, AdjustGamma, Albu, BioMedical3DPad,
|
|
BioMedical3DRandomCrop, BioMedical3DRandomFlip,
|
|
BioMedicalGaussianBlur, BioMedicalGaussianNoise,
|
|
BioMedicalRandomGamma, ConcatCDInput, GenerateEdge,
|
|
LoadAnnotations, LoadBiomedicalAnnotation,
|
|
LoadBiomedicalData, LoadBiomedicalImageFromFile,
|
|
LoadImageFromNDArray, LoadMultipleRSImageFromFile,
|
|
LoadSingleRSImageFromFile, PackSegInputs,
|
|
PhotoMetricDistortion, RandomCrop, RandomCutOut,
|
|
RandomMosaic, RandomRotate, RandomRotFlip, Rerange,
|
|
ResizeShortestEdge, ResizeToMultiple, RGB2Gray,
|
|
SegRescale)
|
|
from .voc import PascalVOCDataset
|
|
|
|
# yapf: enable
|
|
__all__ = [
|
|
'BaseSegDataset', 'BioMedical3DRandomCrop', 'BioMedical3DRandomFlip',
|
|
'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',
|
|
'BioMedicalGaussianNoise', 'BioMedicalGaussianBlur',
|
|
'BioMedicalRandomGamma', 'BioMedical3DPad', 'RandomRotFlip',
|
|
'SynapseDataset', 'REFUGEDataset', 'MapillaryDataset_v1',
|
|
'MapillaryDataset_v2', 'Albu', 'LEVIRCDDataset',
|
|
'LoadMultipleRSImageFromFile', 'LoadSingleRSImageFromFile',
|
|
'ConcatCDInput', 'BaseCDDataset'
|
|
]
|