openocry 0.1.5__py3-none-any.whl

openocr/__init__.py

@@ -0,0 +1,45 @@
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+import os
+import sys
+
+__dir__ = os.path.dirname(os.path.abspath(__file__))
+sys.path.append(__dir__)
+sys.path.insert(0, os.path.abspath(os.path.join(__dir__, '..')))
+
+# from .tools.infer_e2e import OpenOCRE2E, OpenDetector, OpenRecognizer
+# from .tools.infer_unirec_onnx import UniRecONNX
+# from .tools.infer_doc_onnx import OpenDocONNX
+from .openocr import OpenOCR, main
+
+__version__ = '0.1.5'
+
+# Lazy import for demo interfaces to avoid initialization on import
+def launch_openocr_demo(*args, **kwargs):
+    """Launch Gradio OCR demo"""
+    from .demo_gradio import launch_demo
+    return launch_demo(*args, **kwargs)
+
+def launch_unirec_demo(*args, **kwargs):
+    """Launch UniRec demo"""
+    from .demo_unirec import launch_demo
+    return launch_demo(*args, **kwargs)
+
+def launch_opendoc_demo(*args, **kwargs):
+    """Launch OpenDoc demo"""
+    from .demo_opendoc import launch_demo
+    return launch_demo(*args, **kwargs)
+
+__all__ = [
+    'OpenOCRE2E',
+    'OpenDetector',
+    'OpenRecognizer',
+    'UniRecONNX',
+    'OpenDocONNX',
+    'OpenOCR',
+    'main',
+    'launch_openocr_demo',
+    'launch_unirec_demo',
+    'launch_opendoc_demo',
+]

openocr/configs/dataset/rec/evaluation.yaml

@@ -0,0 +1,41 @@
+root: ../evaluation
+task: str
+download_links:
+  # IC15_1811
+  - https://drive.usercontent.google.com/download?id=1eGY0kXNV1qVxeUpoGzs-ioUO-ky7msH6&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1BWv7aLoLAT7avY326gXP3GJF48UZpuBC&authuser=0&confirm=t
+  # SVT
+  - https://drive.usercontent.google.com/download?id=1ecEZ4cJ7dIbTCZRltE0s5KzUotQWagH-&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1OygBP7i9R-3Pwi6WodCcW31J8CUMugOJ&authuser=0&confirm=t
+  # IIIT5k
+  - https://drive.usercontent.google.com/download?id=1PJ9_IvIGZTS5hHdGLnpKuYKZcCO8jE0E&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=10P3MixSBt1v8k8_6aFfziC33Z5IlM6Uf&authuser=0&confirm=t
+  # IC13_857
+  - https://drive.usercontent.google.com/download?id=1-wMHOFBXJaOaY-UD00nDn6qw2s_8R4Vd&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1J1QCFtOFxFKiLJIgTqZ6eRo9Y5QGqHpA&authuser=0&confirm=t
+  # SVTP
+  - https://drive.usercontent.google.com/download?id=1kckwfZkdaHG8k_FW5IIJKUaYZkF21Hza&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1x61lm_ea7lvIdxNPMG-jy-5W0MxtdH0N&authuser=0&confirm=t
+  # CUTE80
+  - https://drive.usercontent.google.com/download?id=1Zv_91c81tinLy5Je89HPr-5wUSnqXKIB&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1OuJ6QoJ9AlyNHIM9j2WedAPxTnac7kyY&authuser=0&confirm=t
+filenames:
+  # IC15_1811
+  - ../evaluation/IC15_1811/data.mdb
+  - ../evaluation/IC15_1811/lock.mdb
+  # SVT
+  - ../evaluation/SVT/data.mdb
+  - ../evaluation/SVT/lock.mdb
+  # IIIT5k
+  - ../evaluation/IIIT5k/data.mdb
+  - ../evaluation/IIIT5k/lock.mdb
+  # IC13_857
+  - ../evaluation/IC13_857/data.mdb
+  - ../evaluation/IC13_857/lock.mdb
+  # SVTP
+  - ../evaluation/SVTP/data.mdb
+  - ../evaluation/SVTP/lock.mdb
+  # CUTE80
+  - ../evaluation/CUTE80/data.mdb
+  - ../evaluation/CUTE80/lock.mdb
+check_validity: true

openocr/configs/dataset/rec/ltb.yaml

@@ -0,0 +1,9 @@
+root: ../ltb
+task: str
+download_links:
+  - https://drive.usercontent.google.com/download?id=16AEA1YGTsyVB44uEjKi4ZUV1snjCYBr4&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1xU4OStrOaI23bPG4flWAPWn2YrQe2bmY&authuser=0&confirm=t
+filenames:
+  - ../ltb/data.mdb
+  - ../ltb/lock.mdb
+check_validity: true

openocr/configs/dataset/rec/mjsynth.yaml

@@ -0,0 +1,11 @@
+root: ../synth
+task: str
+download_links:
+  - https://drive.usercontent.google.com/download?id=1FIoplSFZ-BKQoRDHDXsVMKa844e-K8PD&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1eckTvaeRtlTZvbO2orrVz-cIuIk6i87K&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1PBXTf-2PnmEvJBsqzJqxxRwzhAZGTiMG&authuser=0&confirm=t
+filenames:
+  - ../synth/MJ_train.zip
+  - ../synth/MJ_val.zip
+  - ../synth/MJ_test.zip
+check_validity: true

openocr/configs/dataset/rec/openvino.yaml

@@ -0,0 +1,25 @@
+root: ../OpenVINO
+task: str
+download_links:
+  # train_1
+  - https://drive.usercontent.google.com/download?id=1q23QAIRTyG0t-bBm4aAwRwiqB6VUfphw&authuser=0&confirm=
+  # train_2
+  - https://drive.usercontent.google.com/download?id=1AtbaJljM68cbZqi5lcM92d9VkQUCbSqI&authuser=0&confirm=
+  # train_5
+  - https://drive.usercontent.google.com/download?id=1dejstYnJ8_sESuO_uvwi__jT1B8gPxf3&authuser=0&confirm=t
+  # train_f
+  - https://drive.usercontent.google.com/download?id=1C4akchTc7-yi1OS_sJ3KP693UKcnecke&authuser=0&confirm=t
+  # validation
+  - https://drive.usercontent.google.com/download?id=17TRzSQhuK_juAxAv3KmX0y13pQP2cz6R&authuser=0&confirm=t
+filenames:
+  # train_1
+  - ../OpenVINO/train_1.zip
+  # train_2
+  - ../OpenVINO/train_2.zip
+  # train_5
+  - ../OpenVINO/train_5.zip
+  # train_f
+  - ../OpenVINO/train_f.zip
+  # validation
+  - ../OpenVINO/validation.zip
+check_validity: true

openocr/configs/dataset/rec/ost.yaml

@@ -0,0 +1,17 @@
+root: ../OST
+task: str
+download_links:
+  # OST heavy
+  - https://drive.usercontent.google.com/download?id=1RGpIFbD_SRlrzZFBoVF_LGvetNx1-5pg&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1Th4MfDf44k0EBpIqCLqVoGRu6G-FP1hq&authuser=0&confirm=t
+  # OST weak
+  - https://drive.usercontent.google.com/download?id=1z5CTDJucUnvALG12Q4UXk1DDKJDd8WJn&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1V17TTkX3sjpV7v0km_F2SDCK0tL3k_ls&authuser=0&confirm=t
+filenames:
+  # OST heavy
+  - ../OST/heavy/data.mdb
+  - ../OST/heavy/lock.mdb
+  # OST weak
+  - ../OST/weak/data.mdb
+  - ../OST/weak/lock.mdb
+check_validity: true

openocr/configs/dataset/rec/synthtext.yaml

@@ -0,0 +1,7 @@
+root: ../synth
+task: str
+download_links:
+  - https://drive.usercontent.google.com/download?id=1T-enqkq6_l2HqrsV3da_h0oJ7CUKu_oc&authuser=0&confirm=t
+filenames:
+  - ../synth/ST.zip
+check_validity: true

openocr/configs/dataset/rec/test.yaml

@@ -0,0 +1,77 @@
+root: ../test
+task: str
+download_links:
+  # IC13_857
+  - https://drive.usercontent.google.com/download?id=1PZSCbe6_DI8MlCqCRWXGT2PP92_frIXq&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1qkN7NDg0zUHxUiZHAeEatDTqlsgpFWp3&authuser=0&confirm=t
+  # IC15_2077
+  - https://drive.usercontent.google.com/download?id=1dFkY3DNbr-Mepn3TWBiA9COEJ63fGFcp&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1UvVwLNZ3tS1YdTBa8MulPzjeVezKaDro&authuser=0&confirm=t
+  # SVTP
+  - https://drive.usercontent.google.com/download?id=1aofeerilxJ7J3S7QxuCEXbmXTpz8Xshx&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1rJ1KoO4K_VUxEAUN_bMgBGzK8_JZAAno&authuser=0&confirm=t
+  # IIIT5k
+  - https://drive.usercontent.google.com/download?id=1XFO2M1Kbgwv3-iTNTmhQXAEjNmKYOeoT&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1stwK2hFsyaV7HHsEG9EYgnUQebNb2_nG&authuser=0&confirm=t
+  # COCOv1.4
+  - https://drive.usercontent.google.com/download?id=1Se2QSGS19xx7Gfy-SUdX9mlAOr2eYsfA&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1xvekFi389QfkH7yS0JIVV0QzjhUspjDv&authuser=0&confirm=t
+  # IC15_1811
+  - https://drive.usercontent.google.com/download?id=1pHsw8wrThD9EGEE6AusQLZozefSj4iyR&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1TXZ1qHuKAksaAlvd3qMv4IHKnN-IJW9a&authuser=0&confirm=t
+  # Uber
+  - https://drive.usercontent.google.com/download?id=1L2j6BZeLTGQ1FIl8HB_D3AFiWLltGV5r&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=12DUj28yzLWxFO_gfMfSjTkRujYD5MNEE&authuser=0&confirm=t
+  # IC13_1095
+  - https://drive.usercontent.google.com/download?id=1fu8onMt3Z6fDLNAiHcm-sQ2qCXduE-FU&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1OQAZtLj8U2Cl4L0ErGFsz6vGIVTTWasD&authuser=0&confirm=t
+  # IC13_1015
+  - https://drive.usercontent.google.com/download?id=1mbsfuvWB282HYfn9tbqcj1nUDkLXcSNB&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1QGogU_hV-oN7iY2POutdD2LDcmK6plnV&authuser=0&confirm=t
+  # ArT
+  - https://drive.usercontent.google.com/download?id=1-53knSy-uTSngCG7wyBngVyTuTCmdnWl&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=172EsSaf7BVaB1ORtohi-Jc_8SuUKZGGf&authuser=0&confirm=t
+  # SVT
+  - https://drive.usercontent.google.com/download?id=1p7aVUr9Yr7c4X4YUBvk2-YP28rraHjn9&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1ALmhvSleZ0yf-lcdbQPP3M9Zc3oqnXij&authuser=0&confirm=t
+  # CUTE80
+  - https://drive.usercontent.google.com/download?id=1Ujr4axHKnu54P2rIGUhkjdM6XlhDYrI_&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1DvZi9L3MqjO2zRUyCg3YvP4qMAt2bsme&authuser=0&confirm=t
+filenames:
+  # IC13_857
+  - ../test/IC13_857/data.mdb
+  - ../test/IC13_857/lock.mdb
+  # IC15_2077
+  - ../test/IC15_2077/data.mdb
+  - ../test/IC15_2077/lock.mdb
+  # SVTP
+  - ../test/SVTP/data.mdb
+  - ../test/SVTP/lock.mdb
+  # IIIT5k
+  - ../test/IIIT5k/data.mdb
+  - ../test/IIIT5k/lock.mdb
+  # COCOv1.4
+  - ../test/COCOv1.4/data.mdb
+  - ../test/COCOv1.4/lock.mdb
+  # IC15_1811
+  - ../test/IC15_1811/data.mdb
+  - ../test/IC15_1811/lock.mdb
+  # Uber
+  - ../test/Uber/data.mdb
+  - ../test/Uber/lock.mdb
+  # IC13_1095
+  - ../test/IC13_1095/data.mdb
+  - ../test/IC13_1095/lock.mdb
+  # IC13_1015
+  - ../test/IC13_1015/data.mdb
+  - ../test/IC13_1015/lock.mdb
+  # ArT
+  - ../test/ArT/data.mdb
+  - ../test/ArT/lock.mdb
+  # SVT
+  - ../test/SVT/data.mdb
+  - ../test/SVT/lock.mdb
+  # CUTE80
+  - ../test/CUTE80/data.mdb
+  - ../test/CUTE80/lock.mdb
+check_validity: true

openocr/configs/dataset/rec/textocr.yaml

@@ -0,0 +1,13 @@
+root: ../TextOCR
+task: str
+download_links:
+  # train
+  - https://drive.usercontent.google.com/download?id=1jVjJFno4pnsU0Cp_kn4MIXQrChmELy92&authuser=0&confirm=
+  # val
+  - https://drive.usercontent.google.com/download?id=1ubIRu01MXIek6OvInu-XjaIbw6277-vw&authuser=0&confirm=t
+filenames:
+  # train
+  - ../TextOCR/train.zip
+  # val
+  - ../TextOCR/val.zip
+check_validity: true

openocr/configs/dataset/rec/textocr_horizontal.yaml

@@ -0,0 +1,13 @@
+root: ../TextOCR_horizontal
+task: str
+download_links:
+  # train
+  - https://drive.usercontent.google.com/download?id=1sWH6J11xbjQb8SH7fdG_8mIKVI81ZQy5&authuser=0&confirm=
+  # val
+  - https://drive.usercontent.google.com/download?id=1gIE-AU2o-5hvg288-bjphO6UkI5AEQ2d&authuser=0&confirm=t
+filenames:
+  # train
+  - ../TextOCR_horizontal/train.zip
+  # val
+  - ../TextOCR_horizontal/val.zip
+check_validity: true

openocr/configs/dataset/rec/union14m_b.yaml

@@ -0,0 +1,47 @@
+root: ../u14m
+task: str
+download_links:
+  # artistic
+  - https://drive.usercontent.google.com/download?id=1Je2DTuFHnkXDI99yDnm9Anl5naWaCQwd&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1xtT_Q0juBJUIvAG55qBxoVNNTECd2usZ&authuser=0&confirm=t
+  # contextless
+  - https://drive.usercontent.google.com/download?id=1_0OzyzWhZOmGrHkayFTVrzhrQrNRDRPR&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1PPgC42y3xoM9bR0HQFbDYbcT3PzMdD_y&authuser=0&confirm=t
+  # salient
+  - https://drive.usercontent.google.com/download?id=1tHLMYBmTqRnxvFOTT3dfLfQiundqFWfd&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=13NQgpAtCK0kh9M5E2pAUmKKEp6Qu5Xwj&authuser=0&confirm=t
+  # multi_words
+  - https://drive.usercontent.google.com/download?id=1IlnDKX3V_Vp9gsDGFB0xoqsVLH1vtxUI&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1mFFjC7C0CwevvkwFU9YeVbZBdps_3Qpb&authuser=0&confirm=t
+  # curve
+  - https://drive.usercontent.google.com/download?id=1MxhMd85cmhUtI2lmtXhZQuFk7lav0_fw&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1N03g-4e-kJG2mRvlM0c5TrwWAkd-iG-Q&authuser=0&confirm=t
+  # general
+  - https://drive.usercontent.google.com/download?id=1Oqt7OaycP466NWoDmoJ3FqS8YP3YRgvu&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1K0MrX5eYNt8IIGFHXCwg0_oI5OF5PPFO&authuser=0&confirm=t
+  # multi_oriented
+  - https://drive.usercontent.google.com/download?id=1TKZFcZPVk0ThqfF-AGhJk_OCLg0ykKbv&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1PAoLMUWuR7O2-7XRoKkNzQcSiznErQzD&authuser=0&confirm=t
+filenames:
+  # artistic
+  - ../u14m/artistic/data.mdb
+  - ../u14m/artistic/lock.mdb
+  # contextless
+  - ../u14m/contextless/data.mdb
+  - ../u14m/contextless/lock.mdb
+  # salient
+  - ../u14m/salient/data.mdb
+  - ../u14m/salient/lock.mdb
+  # multi_words
+  - ../u14m/multi_words/data.mdb
+  - ../u14m/multi_words/lock.mdb
+  # curve
+  - ../u14m/curve/data.mdb
+  - ../u14m/curve/lock.mdb
+  # general
+  - ../u14m/general/data.mdb
+  - ../u14m/general/lock.mdb
+  # multi_oriented
+  - ../u14m/multi_oriented/data.mdb
+  - ../u14m/multi_oriented/lock.mdb
+check_validity: true

openocr/configs/dataset/rec/union14m_l_filtered.yaml

@@ -0,0 +1,35 @@
+root: ../Union14M-L-LMDB-Filtered
+task: str
+download_links:
+  # train_challenging
+  - https://drive.usercontent.google.com/download?id=1etwzBgGHjsFsb0sygsaRnKbanW2PMe07&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1ly6FJfPjItwGlVQ-ifTrzzM3rVu3Ezhr&authuser=0&confirm=t
+  # train_easy
+  - https://drive.usercontent.google.com/download?id=1_zeNluTnywIaa5h3PN-Ah9tKyByypot7&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1caYLeQHDidXgVBDi9IWXbO1gg__DYq9a&authuser=0&confirm=t
+  # train_hard
+  - https://drive.usercontent.google.com/download?id=1eP6s2xyYPZX9gykvWA4VSOc3Fqul_UB_&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1-ZlCvocX8P5uVRclUXp_5DNGLDzd16EO&authuser=0&confirm=t
+  # train_medium
+  - https://drive.usercontent.google.com/download?id=1s_CoaLNJEr-UxHYiqZ5jOcliMCFiRUUy&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1Wpj6WVpZ5Ily77kVwfQ18CiZBzkgmEnF&authuser=0&confirm=t
+  # train_normal
+  - https://drive.usercontent.google.com/download?id=1jPt44arlAswl9cXZjzmVcdpptdTPpJ3I&authuser=0&confirm=t
+  - https://drive.usercontent.google.com/download?id=1Rfc5kE03AzOUv7B_eYcBhUV8KMQ2MZ1m&authuser=0&confirm=t
+filenames:
+  # train_challenging
+  - ../Union14M-L-LMDB-Filtered/train_challenging/data.mdb
+  - ../Union14M-L-LMDB-Filtered/train_challenging/lock.mdb
+  # train_easy
+  - ../Union14M-L-LMDB-Filtered/train_easy/data.mdb
+  - ../Union14M-L-LMDB-Filtered/train_easy/lock.mdb
+  # train_hard
+  - ../Union14M-L-LMDB-Filtered/train_hard/data.mdb
+  - ../Union14M-L-LMDB-Filtered/train_hard/lock.mdb
+  # train_medium
+  - ../Union14M-L-LMDB-Filtered/train_medium/data.mdb
+  - ../Union14M-L-LMDB-Filtered/train_medium/lock.mdb
+  # train_normal
+  - ../Union14M-L-LMDB-Filtered/train_normal/data.mdb
+  - ../Union14M-L-LMDB-Filtered/train_normal/lock.mdb
+check_validity: true

openocr/configs/det/dbnet/repvit_db.yml

@@ -0,0 +1,171 @@
+Global:
+  device: gpu
+  epoch_num: &epoch_num 500
+  log_smooth_window: 20
+  print_batch_step: 100
+  output_dir: ./output/det_repsvtr_db
+  save_epoch_step: [400, 25]
+  eval_batch_step:
+  - 0
+  - 1000
+  cal_metric_during_train: false
+  checkpoints:
+  pretrained_model: openocr_det_repvit_ch.pth
+  save_inference_dir: null
+  use_tensorboard: false
+  infer_img:
+  save_res_path: ./checkpoints/det_db/predicts_db.txt
+  distributed: true
+  model_type: det
+
+Architecture:
+  algorithm: DB_mobile
+  Backbone:
+    name: RepSVTR_det
+  Neck:
+    name: RSEFPN
+    out_channels: 96
+    shortcut: True
+  Head:
+    name: DBHead
+    k: 50
+
+Loss:
+  name: DBLoss
+  balance_loss: true
+  main_loss_type: DiceLoss
+  alpha: 5
+  beta: 10
+  ohem_ratio: 3
+
+Optimizer:
+  name: Adam
+  lr: 0.001
+  weight_decay: 5.0e-05
+  filter_bias_and_bn: False
+
+LRScheduler:
+  name: CosineAnnealingLR
+  warmup_epoch: 2
+
+PostProcess:
+  name: DBPostProcess
+  thresh: 0.3
+  box_thresh: 0.6
+  max_candidates: 1000
+  unclip_ratio: 1.5
+  score_mode: 'slow'
+
+Metric:
+  name: DetMetric
+  main_indicator: hmean
+
+Train:
+  dataset:
+    name: SimpleDataSet
+    data_dir: ../icdar2015/text_localization/
+    label_file_list:
+      - ../icdar2015/text_localization/train_icdar2015_label.txt
+    ratio_list: [1.0]
+    transforms:
+    - DecodeImage:
+        img_mode: BGR
+        channel_first: false
+    - DetLabelEncode: null
+    - CopyPaste: null
+    - IaaAugment:
+        augmenter_args:
+        - type: Fliplr
+          args:
+            p: 0.5
+        - type: Affine
+          args:
+            rotate:
+            - -10
+            - 10
+        - type: Resize
+          args:
+            size:
+            - 0.5
+            - 3
+    - EastRandomCropData:
+        size:
+        - 640
+        - 640
+        max_tries: 50
+        keep_ratio: true
+    - MakeBorderMap:
+        shrink_ratio: 0.4
+        thresh_min: 0.3
+        thresh_max: 0.7
+        total_epoch: *epoch_num
+    - MakeShrinkMap:
+        shrink_ratio: 0.4
+        min_text_size: 8
+        total_epoch: *epoch_num
+    - NormalizeImage:
+        scale: 1./255.
+        mean:
+        - 0.485
+        - 0.456
+        - 0.406
+        std:
+        - 0.229
+        - 0.224
+        - 0.225
+        order: hwc
+    - ToCHWImage: null
+    - KeepKeys:
+        keep_keys:
+        - image
+        - threshold_map
+        - threshold_mask
+        - shrink_map
+        - shrink_mask
+  loader:
+    shuffle: true
+    drop_last: false
+    batch_size_per_card: 8
+    num_workers: 8
+
+Eval:
+  dataset:
+    name: SimpleDataSet
+    data_dir: ../icdar2015/text_localization/
+    label_file_list:
+      - ../icdar2015/text_localization/test_icdar2015_label.txt
+    transforms:
+    - DecodeImage:
+        img_mode: BGR
+        channel_first: false
+    - DetLabelEncode: null
+    - DetResizeForTest:
+        # image_shape: [1280, 1280]
+        # keep_ratio: True
+        # padding: True
+        limit_side_len: 960
+        limit_type: max
+    - NormalizeImage:
+        scale: 1./255.
+        mean:
+        - 0.485
+        - 0.456
+        - 0.406
+        std:
+        - 0.229
+        - 0.224
+        - 0.225
+        order: hwc
+    - ToCHWImage: null
+    - KeepKeys:
+        keep_keys:
+        - image
+        - shape
+        - polys
+        - ignore_tags
+  loader:
+    shuffle: false
+    drop_last: false
+    batch_size_per_card: 1
+    num_workers: 2
+profiler_options: null

openocr/configs/rec/abinet/resnet45_trans_abinet_lang.yml

@@ -0,0 +1,94 @@
+Global:
+  device: gpu
+  epoch_num: 20
+  log_smooth_window: 20
+  print_batch_step: 10
+  output_dir: ./output/rec/u14m_filter/resnet45_trans_abinet_lang/
+  eval_epoch_step: [0, 1]
+  eval_batch_step: [0, 500]
+  cal_metric_during_train: True
+  pretrained_model:
+  # ./openocr_nolang_abinet_lang.pth
+  checkpoints:
+  use_tensorboard: false
+  infer_img:
+  # for data or label process
+  character_dict_path: ./tools/utils/EN_symbol_dict.txt
+  max_text_length: 25
+  use_space_char: False
+  save_res_path: ./output/rec/u14m_filter/predicts_resnet45_trans_abinet_lang.txt
+  grad_clip_val: 20
+  use_amp: True
+
+Optimizer:
+  name: Adam
+  lr: 0.000267
+  weight_decay: 0.0
+  filter_bias_and_bn: False
+
+LRScheduler:
+  name: MultiStepLR
+  milestones: [12]
+  gamma: 0.1
+
+Architecture:
+  model_type: rec
+  algorithm: ABINet
+  Transform:
+  Encoder:
+    name: ResNet45
+    in_channels: 3
+    strides: [2, 1, 2, 1, 1]
+  Decoder:
+    name: ABINetDecoder
+    iter_size: 3
+
+Loss:
+  name: ABINetLoss
+
+PostProcess:
+  name: ABINetLabelDecode
+
+Metric:
+  name: RecMetric
+  main_indicator: acc
+  is_filter: True
+
+Train:
+  dataset:
+    name: LMDBDataSet
+    data_dir: ../Union14M-L-LMDB-Filtered
+    transforms:
+      - DecodeImagePIL: # load image
+          img_mode: RGB
+      - PARSeqAugPIL:
+      - ABINetLabelEncode:
+      - RecTVResize:
+          image_shape: [32, 128]
+          padding: False
+      - KeepKeys:
+          keep_keys: ['image', 'label', 'length'] # dataloader will return list in this order
+  loader:
+    shuffle: True
+    batch_size_per_card: 256
+    drop_last: True
+    num_workers: 4
+
+Eval:
+  dataset:
+    name: LMDBDataSet
+    data_dir: ../evaluation
+    transforms:
+      - DecodeImagePIL: # load image
+          img_mode: RGB
+      - ABINetLabelEncode:
+      - RecTVResize:
+          image_shape: [32, 128]
+          padding: False
+      - KeepKeys:
+          keep_keys: ['image', 'label', 'length'] # dataloader will return list in this order
+  loader:
+    shuffle: False
+    drop_last: False
+    batch_size_per_card: 256
+    num_workers: 2