[Transforms] SVTR transforms (#1645)

* rec transforms

* fix

* ut

* update docs

* fix

* new name

* fix

Author: gaotongxiao

docs/en/api/transforms.rst

@@ -34,7 +34,6 @@ TextDet Transforms
    :template: classtemplate.rst
 
    BoundedScaleAspectJitter
-   FixInvalidPolygon
    RandomFlip
    SourceImagePad
    ShortScaleAspectJitter
@@ -50,6 +49,10 @@ TextRecog Transforms
    :nosignatures:
    :template: classtemplate.rst
 
+   TextRecogGeneralAug
+   CropHeight
+   ImageContentJitter
+   ReversePixels
    PyramidRescale
    PadToWidth
    RescaleToHeight
@@ -66,6 +69,8 @@ OCR Transforms
    RandomCrop
    RandomRotate
    Resize
+   FixInvalidPolygon
+   RemoveIgnored
 
 
 

docs/zh_cn/api/transforms.rst

@@ -34,7 +34,6 @@ TextDet Transforms
    :template: classtemplate.rst
 
    BoundedScaleAspectJitter
-   FixInvalidPolygon
    RandomFlip
    SourceImagePad
    ShortScaleAspectJitter
@@ -50,6 +49,10 @@ TextRecog Transforms
    :nosignatures:
    :template: classtemplate.rst
 
+   TextRecogGeneralAug
+   CropHeight
+   ImageContentJitter
+   ReversePixels
    PyramidRescale
    PadToWidth
    RescaleToHeight
@@ -66,6 +69,8 @@ OCR Transforms
    RandomCrop
    RandomRotate
    Resize
+   FixInvalidPolygon
+   RemoveIgnored
 
 
 

mmocr/datasets/transforms/__init__.py

@@ -9,7 +9,9 @@
 from .textdet_transforms import (BoundedScaleAspectJitter, RandomFlip,
                                  ShortScaleAspectJitter, SourceImagePad,
                                  TextDetRandomCrop, TextDetRandomCropFlip)
-from .textrecog_transforms import PadToWidth, PyramidRescale, RescaleToHeight
+from .textrecog_transforms import (CropHeight, ImageContentJitter, PadToWidth,
+                                   PyramidRescale, RescaleToHeight,
+                                   ReversePixels, TextRecogGeneralAug)
 from .wrappers import ConditionApply, ImgAugWrapper, TorchVisionWrapper
 
 __all__ = [
@@ -20,5 +22,6 @@
     'ShortScaleAspectJitter', 'RandomFlip', 'BoundedScaleAspectJitter',
     'PackKIEInputs', 'LoadKIEAnnotations', 'FixInvalidPolygon', 'MMDet2MMOCR',
     'MMOCR2MMDet', 'LoadImageFromLMDB', 'LoadImageFromFile',
-    'LoadImageFromNDArray', 'RemoveIgnored', 'ConditionApply'
+    'LoadImageFromNDArray', 'CropHeight', 'TextRecogGeneralAug',
+    'ImageContentJitter', 'ReversePixels', 'RemoveIgnored', 'ConditionApply'
 ]