mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
6af2b8eab9
## Motivation Add the 3d pad transform for biomedical images, which follows the design of the nnUNet.
51 lines
2.4 KiB
Python
51 lines
2.4 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 .stare import STAREDataset
|
|
# yapf: disable
|
|
from .transforms import (CLAHE, AdjustGamma, BioMedical3DPad,
|
|
BioMedical3DRandomCrop, BioMedicalGaussianBlur,
|
|
BioMedicalGaussianNoise, BioMedicalRandomGamma,
|
|
GenerateEdge, LoadAnnotations,
|
|
LoadBiomedicalAnnotation, LoadBiomedicalData,
|
|
LoadBiomedicalImageFromFile, LoadImageFromNDArray,
|
|
PackSegInputs, PhotoMetricDistortion, RandomCrop,
|
|
RandomCutOut, RandomMosaic, RandomRotate, Rerange,
|
|
ResizeShortestEdge, ResizeToMultiple, RGB2Gray,
|
|
SegRescale)
|
|
from .voc import PascalVOCDataset
|
|
|
|
# yapf: enable
|
|
__all__ = [
|
|
'BaseSegDataset', 'BioMedical3DRandomCrop', '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'
|
|
]
|