瀏覽代碼

optimize layout model setting in layout_parsing_v2

1. set layout_merge_bboxes_mode according to class;
1. set threshold of seal to 0.3;
gaotingquan 8 月之前
父節點
當前提交
28f6e4f732
共有 1 個文件被更改,包括 6 次插入4 次删除
  1. 6 4
      paddlex/configs/pipelines/layout_parsing_v2.yaml

+ 6 - 4
paddlex/configs/pipelines/layout_parsing_v2.yaml

@@ -13,12 +13,14 @@ SubModules:
     model_name: PP-DocLayout-L
     model_dir: null
     threshold: 
-      0: 0.3
-      7: 0.3
-      16: 0.3
+      0: 0.3  # paragraph_title
+      7: 0.3  # formula
+      16: 0.3 # seal
     layout_nms: True
     layout_unclip_ratio: 1.0
-    layout_merge_bboxes_mode: "union"
+    layout_merge_bboxes_mode: 
+      1: "large"  # image
+      18: "large" # chart
 
 SubPipelines:
   DocPreprocessor: