PaddleX/paddlex/inference/pipelines/layout_parsing/setting.py

# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
#
# 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.

parameters_config = {
    "page": {},
    "region": {
        "match_block_overlap_ratio_threshold": 0.8,
        "split_block_overlap_ratio_threshold": 0.4,
    },
    "block": {
        "title_conversion_area_ratio_threshold": 0.3,  # update paragraph_title -> doc_title
    },
    "line": {
        "line_height_iou_threshold": 0.6,  # For line segmentation of OCR results
        "delimiter_map": {
            "doc_title": " ",
            "content": "\n",
        },
    },
    "word": {
        "delimiter": " ",
    },
    "order": {
        "block_label_match_iou_threshold": 0.1,
        "block_title_match_iou_threshold": 0.1,
    },
}

block_label_mapping = {
    "doc_title_labels": ["doc_title"],  # 文档标题
    "paragraph_title_labels": [
        "paragraph_title",
        "abstract_title",
        "reference_title",
        "content_title",
    ],  # 段落标题
    "vision_labels": [
        "image",
        "table",
        "chart",
        "flowchart",
        "figure",
    ],  # 图、表、印章、图表、图
    "vision_title_labels": ["table_title", "chart_title", "figure_title"],  # 图表标题
    "unordered_labels": [
        "aside_text",
        "seal",
        "number",
        "formula_number",
    ],
    "text_labels": ["text"],
    "header_labels": ["header", "header_image"],
    "footer_labels": ["footer", "footer_image", "footnote"],
    "visualize_index_labels": [
        "text",
        "formula",
        "algorithm",
        "reference",
        "content",
        "abstract",
        "paragraph_title",
        "doc_title",
        "table_title",
        "chart_title",
        "figure_title",
        "image",
        "table",
        "chart",
        "figure",
        "abstract_title",
        "refer_title",
        "content_title",
        "flowchart",
    ],
}