Echo/fcb_photo_review
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

现阶段布局分析可能会反优化，暂不开启

Browse Source
This commit is contained in:
liuyebo
2024-06-18 17:16:24 +08:00
parent 2eb35b637f
commit 2e03863fc8
3 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
visual_model_test/visual_model_test.py
View File

@@ -44,7 +44,7 @@ def visual_model_test(model_type, test_img, task_path, schema):
write_visual_result(test_img, layout=parsed_doc["layout"])
else:
my_ie = Taskflow("information_extraction", schema=schema, model="uie-x-base", task_path=task_path,
layout_analysis=True)
layout_analysis=False)
my_results = my_ie({"doc": test_img})
write_visual_result(test_img, result=my_results[0])
@@ -74,7 +74,7 @@ def main(model_type, pic_name=None):
schema = ["患者姓名", "入院日期", "出院日期", "费用总额", "个人现金支付", "个人账户支付", "自费金额", "医保类型"]
elif model_type == "discharge":
task_path = "../config/model/discharge_record_model"
test_img_path = ucloud.get_private_url(pic_name) if pic_name else "img/PH20240428000832_1_093844_2.jpg"
test_img_path = ucloud.get_private_url(pic_name) if pic_name else "img/PH20240401000003_3_001938_2.jpg"
schema = ["医院", "科室", "患者姓名", "入院日期", "出院日期", "主治医生"]
elif model_type == "cost":
task_path = "../config/model/cost_list_model"