deepdoctection / conf_dd_one.yaml
JaMe76's picture
Upload conf_dd_one.yaml
b958717
raw history blame
No virus
693 Bytes
CONFIG:
D2LAYOUT: dd/d2/layout/CASCADE_RCNN_R_50_FPN_GN.yaml
D2CELL: dd/d2/cell/CASCADE_RCNN_R_50_FPN_GN.yaml
D2ITEM: dd/d2/item/CASCADE_RCNN_R_50_FPN_GN.yaml
WEIGHTS:
D2LAYOUT: layout/model_final_inf_only.pt
D2CELL: cell/d2_model_1849999_cell_inf_only.pt
D2ITEM: item/d2_model_1639999_item_inf_only.pt
SEGMENTATION:
ASSIGNMENT_RULE: ioa
IOU_THRESHOLD_ROWS: 0.01
IOU_THRESHOLD_COLS: 0.001
IOA_THRESHOLD_ROWS: 0.4
IOA_THRESHOLD_COLS: 0.4
FULL_TABLE_TILING: True
REMOVE_IOU_THRESHOLD_ROWS: 0.001
REMOVE_IOU_THRESHOLD_COLS: 0.001
WORD_MATCHING:
PARENTAL_CATEGORIES:
- TEXT
- TITLE
- CELL
- LIST
RULE: ioa
IOU_THRESHOLD: 0.001
IOA_THRESHOLD: 0.6