DeepOCR / conf_dd_one.yaml
Arnaudding001's picture
Create conf_dd_one.yaml
9748f23
raw
history blame contribute delete
694 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