external/det/mmdet/datasets/transforms/__init__.py

# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.

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 .transformers_glip import GTBoxSubOne_GLIP, RandomFlip_GLIP
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', 'GTBoxSubOne_GLIP', 'RandomFlip_GLIP'
]