mmdetection/mmdet/datasets/transforms/__init__.py

# Copyright (c) OpenMMLab. All rights reserved.
from .augment_wrappers import AutoAugment, RandAugment
from .colorspace import (AutoContrast, Brightness, Color, ColorTransform,
                         Contrast, Equalize, Invert, Posterize, Sharpness,
                         Solarize, SolarizeAdd)
from .formatting import (ImageToTensor, PackDetInputs, PackReIDInputs,
                         PackTrackInputs, ToTensor, Transpose)
from .frame_sampling import BaseFrameSample, UniformRefFrameSample
from .geometric import (GeomTransform, Rotate, ShearX, ShearY, TranslateX,
                        TranslateY)
from .instaboost import InstaBoost
from .loading import (FilterAnnotations, InferencerLoader, LoadAnnotations,
                      LoadEmptyAnnotations, LoadImageFromNDArray,
                      LoadMultiChannelImageFromFiles, LoadPanopticAnnotations,
                      LoadProposals, LoadTrackAnnotations)
from .transforms import (Albu, CachedMixUp, CachedMosaic, CopyPaste, CutOut,
                         Expand, FixScaleResize, FixShapeResize,
                         MinIoURandomCrop, MixUp, Mosaic, Pad,
                         PhotoMetricDistortion, RandomAffine,
                         RandomCenterCropPad, RandomCrop, RandomErasing,
                         RandomFlip, RandomShift, Resize, ResizeShortestEdge,
                         SegRescale, YOLOXHSVRandomAug)
from .wrappers import MultiBranch, ProposalBroadcaster, RandomOrder

__all__ = [
    'PackDetInputs', 'ToTensor', 'ImageToTensor', 'Transpose',
    'LoadImageFromNDArray', 'LoadAnnotations', 'LoadPanopticAnnotations',
    'LoadMultiChannelImageFromFiles', 'LoadProposals', 'Resize', 'RandomFlip',
    'RandomCrop', 'SegRescale', 'MinIoURandomCrop', 'Expand',
    'PhotoMetricDistortion', 'Albu', 'InstaBoost', 'RandomCenterCropPad',
    'AutoAugment', 'CutOut', 'ShearX', 'ShearY', 'Rotate', 'Color', 'Equalize',
    'Brightness', 'Contrast', 'TranslateX', 'TranslateY', 'RandomShift',
    'Mosaic', 'MixUp', 'RandomAffine', 'YOLOXHSVRandomAug', 'CopyPaste',
    'FilterAnnotations', 'Pad', 'GeomTransform', 'ColorTransform',
    'RandAugment', 'Sharpness', 'Solarize', 'SolarizeAdd', 'Posterize',
    'AutoContrast', 'Invert', 'MultiBranch', 'RandomErasing',
    'LoadEmptyAnnotations', 'RandomOrder', 'CachedMosaic', 'CachedMixUp',
    'FixShapeResize', 'ProposalBroadcaster', 'InferencerLoader',
    'LoadTrackAnnotations', 'BaseFrameSample', 'UniformRefFrameSample',
    'PackTrackInputs', 'PackReIDInputs', 'FixScaleResize', 'ResizeShortestEdge'
]