mmocr/datasets/pipelines/__init__.py

# Copyright (c) OpenMMLab. All rights reserved.
from .box_utils import sort_vertex, sort_vertex8
from .custom_format_bundle import CustomFormatBundle
from .dbnet_transforms import EastRandomCrop, ImgAug
from .kie_transforms import KIEFormatBundle, ResizeNoImg
from .loading import LoadImageFromNdarray, LoadTextAnnotations
from .ner_transforms import NerTransform, ToTensorNER
from .ocr_seg_targets import OCRSegTargets
from .ocr_transforms import (FancyPCA, NormalizeOCR, OnlineCropOCR,
                             OpencvToPil, PilToOpencv, RandomPaddingOCR,
                             RandomRotateImageBox, ResizeOCR, ToTensorOCR)
from .test_time_aug import MultiRotateAugOCR
from .textdet_targets import (DBNetTargets, FCENetTargets, PANetTargets,
                              TextSnakeTargets)
from .transform_wrappers import OneOfWrapper, RandomWrapper, TorchVisionWrapper
from .transforms import (ColorJitter, PyramidRescale, RandomCropFlip,
                         RandomCropInstances, RandomCropPolyInstances,
                         RandomRotatePolyInstances, RandomRotateTextDet,
                         RandomScaling, ScaleAspectJitter, SquareResizePad)

__all__ = [
    'LoadTextAnnotations', 'NormalizeOCR', 'OnlineCropOCR', 'ResizeOCR',
    'ToTensorOCR', 'CustomFormatBundle', 'DBNetTargets', 'PANetTargets',
    'ColorJitter', 'RandomCropInstances', 'RandomRotateTextDet',
    'ScaleAspectJitter', 'MultiRotateAugOCR', 'OCRSegTargets', 'FancyPCA',
    'RandomCropPolyInstances', 'RandomRotatePolyInstances', 'RandomPaddingOCR',
    'ImgAug', 'EastRandomCrop', 'RandomRotateImageBox', 'OpencvToPil',
    'PilToOpencv', 'KIEFormatBundle', 'SquareResizePad', 'TextSnakeTargets',
    'sort_vertex', 'LoadImageFromNdarray', 'sort_vertex8', 'FCENetTargets',
    'RandomScaling', 'RandomCropFlip', 'NerTransform', 'ToTensorNER',
    'ResizeNoImg', 'PyramidRescale', 'OneOfWrapper', 'RandomWrapper',
    'TorchVisionWrapper'
]