first commit

2025-07-02 08:57:16 +03:00
commit 56532cc9a9
1901 changed files with 457695 additions and 0 deletions
--- a/ppocr/postprocess/init.py
+++ b/ppocr/postprocess/init.py
@@ -0,0 +1,121 @@
+# copyright (c) 2020 PaddlePaddle Authors. All Rights Reserve.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+from __future__ import unicode_literals
+
+import os
+import copy
+
+__all__ = ["build_post_process"]
+
+from .db_postprocess import DBPostProcess, DistillationDBPostProcess
+from .east_postprocess import EASTPostProcess
+from .sast_postprocess import SASTPostProcess
+from .fce_postprocess import FCEPostProcess
+from .rec_postprocess import (
+    CTCLabelDecode,
+    AttnLabelDecode,
+    SRNLabelDecode,
+    DistillationCTCLabelDecode,
+    NRTRLabelDecode,
+    SARLabelDecode,
+    SEEDLabelDecode,
+    PRENLabelDecode,
+    ViTSTRLabelDecode,
+    ABINetLabelDecode,
+    SPINLabelDecode,
+    VLLabelDecode,
+    RFLLabelDecode,
+    SATRNLabelDecode,
+    ParseQLabelDecode,
+    CPPDLabelDecode,
+    LaTeXOCRDecode,
+    UniMERNetDecode,
+)
+from .cls_postprocess import ClsPostProcess
+from .pg_postprocess import PGPostProcess
+from .vqa_token_ser_layoutlm_postprocess import (
+    VQASerTokenLayoutLMPostProcess,
+    DistillationSerPostProcess,
+)
+from .vqa_token_re_layoutlm_postprocess import (
+    VQAReTokenLayoutLMPostProcess,
+    DistillationRePostProcess,
+)
+from .table_postprocess import TableMasterLabelDecode, TableLabelDecode
+from .picodet_postprocess import PicoDetPostProcess
+from .ct_postprocess import CTPostProcess
+from .drrg_postprocess import DRRGPostprocess
+from .rec_postprocess import CANLabelDecode
+
+
+def build_post_process(config, global_config=None):
+    support_dict = [
+        "DBPostProcess",
+        "EASTPostProcess",
+        "SASTPostProcess",
+        "FCEPostProcess",
+        "CTCLabelDecode",
+        "AttnLabelDecode",
+        "ClsPostProcess",
+        "SRNLabelDecode",
+        "PGPostProcess",
+        "DistillationCTCLabelDecode",
+        "TableLabelDecode",
+        "DistillationDBPostProcess",
+        "NRTRLabelDecode",
+        "SARLabelDecode",
+        "SEEDLabelDecode",
+        "VQASerTokenLayoutLMPostProcess",
+        "VQAReTokenLayoutLMPostProcess",
+        "PRENLabelDecode",
+        "DistillationSARLabelDecode",
+        "ViTSTRLabelDecode",
+        "ABINetLabelDecode",
+        "TableMasterLabelDecode",
+        "SPINLabelDecode",
+        "DistillationSerPostProcess",
+        "DistillationRePostProcess",
+        "VLLabelDecode",
+        "PicoDetPostProcess",
+        "CTPostProcess",
+        "RFLLabelDecode",
+        "DRRGPostprocess",
+        "CANLabelDecode",
+        "SATRNLabelDecode",
+        "ParseQLabelDecode",
+        "CPPDLabelDecode",
+        "LaTeXOCRDecode",
+        "UniMERNetDecode",
+    ]
+
+    if config["name"] == "PSEPostProcess":
+        from .pse_postprocess import PSEPostProcess
+
+        support_dict.append("PSEPostProcess")
+
+    config = copy.deepcopy(config)
+    module_name = config.pop("name")
+    if module_name == "None":
+        return
+    if global_config is not None:
+        config.update(global_config)
+    assert module_name in support_dict, Exception(
+        "post process only support {}".format(support_dict)
+    )
+    module_class = eval(module_name)(**config)
+    return module_class