mmocr/datasets/transforms/__init__.py

# Copyright (c) OpenMMLab. All rights reserved.
from .adapters import MMDet2MMOCR, MMOCR2MMDet
from .formatting import PackKIEInputs, PackTextDetInputs, PackTextRecogInputs
from .loading import (InferencerLoader, LoadImageFromFile,
                      LoadImageFromNDArray, LoadKIEAnnotations,
                      LoadOCRAnnotations)
from .ocr_transforms import (FixInvalidPolygon, RandomCrop, RandomRotate,
                             RemoveIgnored, Resize)
from .textdet_transforms import (BoundedScaleAspectJitter, RandomFlip,
                                 ShortScaleAspectJitter, SourceImagePad,
                                 TextDetRandomCrop, TextDetRandomCropFlip)
from .textrecog_transforms import (CropHeight, ImageContentJitter, PadToWidth,
                                   PyramidRescale, RescaleToHeight,
                                   ReversePixels, TextRecogGeneralAug)
from .wrappers import ConditionApply, ImgAugWrapper, TorchVisionWrapper

__all__ = [
    'LoadOCRAnnotations', 'RandomRotate', 'ImgAugWrapper', 'SourceImagePad',
    'TextDetRandomCropFlip', 'PyramidRescale', 'TorchVisionWrapper', 'Resize',
    'RandomCrop', 'TextDetRandomCrop', 'RandomCrop', 'PackTextDetInputs',
    'PackTextRecogInputs', 'RescaleToHeight', 'PadToWidth',
    'ShortScaleAspectJitter', 'RandomFlip', 'BoundedScaleAspectJitter',
    'PackKIEInputs', 'LoadKIEAnnotations', 'FixInvalidPolygon', 'MMDet2MMOCR',
    'MMOCR2MMDet', 'LoadImageFromFile', 'LoadImageFromNDArray', 'CropHeight',
    'InferencerLoader', 'RemoveIgnored', 'ConditionApply', 'CropHeight',
    'TextRecogGeneralAug', 'ImageContentJitter', 'ReversePixels'
]