54 lines
2.6 KiB
Python
54 lines
2.6 KiB
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
# yapf: disable
|
|
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 .refuge import REFUGEDataset
|
|
from .stare import STAREDataset
|
|
from .synapse import SynapseDataset
|
|
# yapf: disable
|
|
from .transforms import (CLAHE, AdjustGamma, BioMedical3DPad,
|
|
BioMedical3DRandomCrop, BioMedical3DRandomFlip,
|
|
BioMedicalGaussianBlur, BioMedicalGaussianNoise,
|
|
BioMedicalRandomGamma, GenerateEdge, LoadAnnotations,
|
|
LoadBiomedicalAnnotation, LoadBiomedicalData,
|
|
LoadBiomedicalImageFromFile, LoadImageFromNDArray,
|
|
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'
|
|
]
|