17 lines
752 B
YAML
17 lines
752 B
YAML
|
detector: "/models/Kie_invoice_ap/wild_receipt_finetune_weights_c_lite.pth"
|
||
|
|
rotator_version: "/models/Kie_invoice_ap/best_bbox_mAP_epoch_30_lite.pth"
|
||
|
|
recog_max_seq_len: 25
|
||
|
|
recognizer: "satrn-lite-general-pretrain-20230106"
|
||
|
|
device: "cuda:0"
|
||
|
|
do_extend_bbox: True
|
||
|
|
margin_bbox: [0, 0.03, 0.02, 0.05]
|
||
|
|
batch_mode: False
|
||
|
|
batch_size: 16
|
||
|
|
auto_rotate: True
|
||
|
|
img_size: [] #[1920,1920] #text det default size: 1280x1280 #[] = originla size, TODO: fix the deskew code to resize the image only for detecting the angle, we want to feed the original size image to the text detection pipeline so that the bounding boxes would be mapped back to the original size
|
||
|
|
deskew: False #True
|
||
|
|
words_to_lines: {
|
||
|
|
"gradient": 0.6,
|
||
|
|
"max_x_dist": 20,
|
||
|
|
"y_overlap_threshold": 0.5,
|
||
|
|
}
|