[Transforms] SVTR transforms (#1645)

* rec transforms * fix * ut * update docs * fix * new name * fix
open-mmlab · Jan 6, 2023 · b0557c2 · b0557c2
1 parent d679691
commit b0557c2
Show file tree

Hide file tree

Showing 5 changed files with 576 additions and 7 deletions.
diff --git a/docs/en/api/transforms.rst b/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
 
 
 

diff --git a/docs/zh_cn/api/transforms.rst b/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
 
 
 

diff --git a/mmocr/datasets/transforms/__init__.py b/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'
 ]