mmsegmentation/mmseg/datasets/pipelines/__init__.py

22 lines
1.0 KiB
Python

# Copyright (c) OpenMMLab. All rights reserved.
from mmcv.transforms import (LoadImageFromFile, MultiScaleFlipAug, Normalize,
Pad, RandomChoiceResize, RandomFlip, RandomResize,
Resize)
from .compose import Compose
from .formatting import (ImageToTensor, PackSegInputs, ToDataContainer,
Transpose)
from .loading import LoadAnnotations
from .transforms import (CLAHE, AdjustGamma, PhotoMetricDistortion, RandomCrop,
RandomCutOut, RandomMosaic, RandomRotate, Rerange,
RGB2Gray, SegRescale)
__all__ = [
'Compose', 'ImageToTensor', 'ToDataContainer', 'Transpose',
'LoadAnnotations', 'LoadImageFromFile', 'RandomFlip', 'Pad', 'RandomCrop',
'Normalize', 'SegRescale', 'PhotoMetricDistortion', 'RandomRotate',
'AdjustGamma', 'CLAHE', 'Rerange', 'RGB2Gray', 'RandomCutOut',
'RandomMosaic', 'PackSegInputs', 'Resize', 'RandomResize',
'RandomChoiceResize', 'MultiScaleFlipAug'
]