Arnaudding001 commited on
Commit
9748f23
1 Parent(s): a5e5649

Create conf_dd_one.yaml

Browse files
Files changed (1) hide show
  1. conf_dd_one.yaml +26 -0
conf_dd_one.yaml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ CONFIG:
2
+ D2LAYOUT: dd/d2/layout/CASCADE_RCNN_R_50_FPN_GN.yaml
3
+ D2CELL: dd/d2/cell/CASCADE_RCNN_R_50_FPN_GN.yaml
4
+ D2ITEM: dd/d2/item/CASCADE_RCNN_R_50_FPN_GN.yaml
5
+ WEIGHTS:
6
+ D2LAYOUT: layout/model_final_inf_only.pt
7
+ D2CELL: cell/d2_model_1849999_cell_inf_only.pt
8
+ D2ITEM: item/d2_model_1639999_item_inf_only.pt
9
+ SEGMENTATION:
10
+ ASSIGNMENT_RULE: ioa
11
+ IOU_THRESHOLD_ROWS: 0.01
12
+ IOU_THRESHOLD_COLS: 0.001
13
+ IOA_THRESHOLD_ROWS: 0.4
14
+ IOA_THRESHOLD_COLS: 0.4
15
+ FULL_TABLE_TILING: True
16
+ REMOVE_IOU_THRESHOLD_ROWS: 0.001
17
+ REMOVE_IOU_THRESHOLD_COLS: 0.001
18
+ WORD_MATCHING:
19
+ PARENTAL_CATEGORIES:
20
+ - TEXT
21
+ - TITLE
22
+ - CELL
23
+ - LIST
24
+ RULE: ioa
25
+ IOU_THRESHOLD: 0.001
26
+ IOA_THRESHOLD: 0.6