Sunflower7788 9 сар өмнө
parent
commit
6293781bb6

+ 75 - 0
paddlex/repo_apis/PaddleDetection_api/object_det/official_categories.py

@@ -142,4 +142,79 @@ official_categories = {
     "PP-YOLOE_plus-S_face": [{"name": "face", "id": 0}],
     "RT-DETR-L_wired_table_cell_det": [{"name": "cell", "id": 0}],
     "RT-DETR-L_wireless_table_cell_det": [{"name": "cell", "id": 0}],
+    "PP-DocLayout-L": [
+        {"name": "paragraph_title", "id": 0},
+        {"name": "image", "id": 1},
+        {"name": "text", "id": 2},
+        {"name": "number", "id": 3},
+        {"name": "abstract", "id": 4},
+        {"name": "content", "id": 5},
+        {"name": "figure_title", "id": 6},
+        {"name": "formula", "id": 7},
+        {"name": "table", "id": 8},
+        {"name": "table_title", "id": 9},
+        {"name": "reference", "id": 10},
+        {"name": "doc_title", "id": 11},
+        {"name": "footnote", "id": 12},
+        {"name": "header", "id": 13},
+        {"name": "algorithm", "id": 14},
+        {"name": "footer", "id": 15},
+        {"name": "seal", "id": 16},
+        {"name": "chart_title", "id": 17},
+        {"name": "chart", "id": 18},
+        {"name": "formula_number", "id": 19},
+        {"name": "header_image", "id": 20},
+        {"name": "footer_image", "id": 21},
+        {"name": "aside_text", "id": 22},
+    ],
+    "PP-DocLayout-M": [
+        {"name": "paragraph_title", "id": 0},
+        {"name": "image", "id": 1},
+        {"name": "text", "id": 2},
+        {"name": "number", "id": 3},
+        {"name": "abstract", "id": 4},
+        {"name": "content", "id": 5},
+        {"name": "figure_title", "id": 6},
+        {"name": "formula", "id": 7},
+        {"name": "table", "id": 8},
+        {"name": "table_title", "id": 9},
+        {"name": "reference", "id": 10},
+        {"name": "doc_title", "id": 11},
+        {"name": "footnote", "id": 12},
+        {"name": "header", "id": 13},
+        {"name": "algorithm", "id": 14},
+        {"name": "footer", "id": 15},
+        {"name": "seal", "id": 16},
+        {"name": "chart_title", "id": 17},
+        {"name": "chart", "id": 18},
+        {"name": "formula_number", "id": 19},
+        {"name": "header_image", "id": 20},
+        {"name": "footer_image", "id": 21},
+        {"name": "aside_text", "id": 22},
+    ],
+    "PP-DocLayout-S": [
+        {"name": "paragraph_title", "id": 0},
+        {"name": "image", "id": 1},
+        {"name": "text", "id": 2},
+        {"name": "number", "id": 3},
+        {"name": "abstract", "id": 4},
+        {"name": "content", "id": 5},
+        {"name": "figure_title", "id": 6},
+        {"name": "formula", "id": 7},
+        {"name": "table", "id": 8},
+        {"name": "table_title", "id": 9},
+        {"name": "reference", "id": 10},
+        {"name": "doc_title", "id": 11},
+        {"name": "footnote", "id": 12},
+        {"name": "header", "id": 13},
+        {"name": "algorithm", "id": 14},
+        {"name": "footer", "id": 15},
+        {"name": "seal", "id": 16},
+        {"name": "chart_title", "id": 17},
+        {"name": "chart", "id": 18},
+        {"name": "formula_number", "id": 19},
+        {"name": "header_image", "id": 20},
+        {"name": "footer_image", "id": 21},
+        {"name": "aside_text", "id": 22},
+    ],
 }