72 lines
1.1 KiB
Python
Executable File
72 lines
1.1 KiB
Python
Executable File
MAX_SEQ_LENGTH = 512
|
|
DEVICE = "cuda:0"
|
|
KIE_LABELS = [
|
|
"other",
|
|
"form_key",
|
|
"form_value",
|
|
"serial_key",
|
|
"serial_value",
|
|
"no_key",
|
|
"no_value",
|
|
"date",
|
|
"seller_name_value",
|
|
"seller_name_key",
|
|
"seller_tax_code_key",
|
|
"seller_tax_code_value",
|
|
"seller_address_value",
|
|
"seller_address_key",
|
|
"seller_mobile_key",
|
|
"buyer_name_key",
|
|
"buyer_company_name_key",
|
|
"buyer_company_name_value",
|
|
"buyer_tax_code_key",
|
|
"buyer_tax_code_value",
|
|
"buyer_address_value",
|
|
"buyer_address_key",
|
|
"VAT_amount_key",
|
|
"VAT_amount_value",
|
|
"total_key",
|
|
"total_value",
|
|
"total_in_words_key",
|
|
"total_in_words_value",
|
|
"seller_mobile_value",
|
|
"buyer_name_value",
|
|
"buyer_mobile_key",
|
|
"buyer_mobile_value",
|
|
]
|
|
|
|
BRIEF_LABELS = [
|
|
"o",
|
|
"fk",
|
|
"fv",
|
|
"sk",
|
|
"sv",
|
|
"nk",
|
|
"nv",
|
|
"d",
|
|
"snv",
|
|
"snk",
|
|
"stck",
|
|
"stcv",
|
|
"sav",
|
|
"sak",
|
|
"smk",
|
|
"bnk",
|
|
"bcnk",
|
|
"bcnv",
|
|
"btck",
|
|
"btcv",
|
|
"bav",
|
|
"bak",
|
|
"VATk",
|
|
"VATv",
|
|
"tk",
|
|
"tv",
|
|
"tiwk",
|
|
"tiwv",
|
|
"smv",
|
|
"bnv",
|
|
"bmk",
|
|
"bmv",
|
|
]
|