17 lines
752 B
YAML
Executable File
17 lines
752 B
YAML
Executable File
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,
|
|
} |