[Feature] Add svtr encoder (#1483)

* add svtr backbone * update backbone * fix * apply comments, move backbone to encoder Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
open-mmlab · Jan 4, 2023 · 5dbacfe · 5dbacfe
1 parent 65e746e
commit 5dbacfe
Show file tree

Hide file tree

Showing 3 changed files with 720 additions and 1 deletion.
diff --git a/mmocr/models/textrecog/encoders/__init__.py b/mmocr/models/textrecog/encoders/__init__.py
@@ -6,8 +6,9 @@
 from .nrtr_encoder import NRTREncoder
 from .sar_encoder import SAREncoder
 from .satrn_encoder import SATRNEncoder
+from .svtr_encoder import SVTREncoder
 
 __all__ = [
     'SAREncoder', 'NRTREncoder', 'BaseEncoder', 'ChannelReductionEncoder',
-    'SATRNEncoder', 'ABIEncoder', 'ASTEREncoder'
+    'SATRNEncoder', 'ABIEncoder', 'SVTREncoder', 'ASTEREncoder'
 ]