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

尝试使用cpu进行识别

Browse Source
This commit is contained in:
liuyebo
2024-07-01 11:10:06 +08:00
parent d91dd2b770
commit ab72f21eff
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
photo_review/photo_review.py
View File

@@ -395,11 +395,12 @@ def main():
if phhds: if phhds:
ie_task_flows = [ ie_task_flows = [
Taskflow("information_extraction", schema=SETTLEMENT_LIST_SCHEMA, model="uie-x-base", Taskflow("information_extraction", schema=SETTLEMENT_LIST_SCHEMA, model="uie-x-base",
task_path="config/model/settlement_list_model", layout_analysis=LAYOUT_ANALYSIS), task_path="config/model/settlement_list_model", layout_analysis=LAYOUT_ANALYSIS, device_id=-1),
Taskflow("information_extraction", schema=DISCHARGE_RECORD_SCHEMA, model="uie-x-base", Taskflow("information_extraction", schema=DISCHARGE_RECORD_SCHEMA, model="uie-x-base",
task_path="config/model/discharge_record_model", layout_analysis=LAYOUT_ANALYSIS), task_path="config/model/discharge_record_model", layout_analysis=LAYOUT_ANALYSIS,
device_id=-1),
Taskflow("information_extraction", schema=COST_LIST_SCHEMA, model="uie-x-base", Taskflow("information_extraction", schema=COST_LIST_SCHEMA, model="uie-x-base",
task_path="config/model/cost_list_model", layout_analysis=LAYOUT_ANALYSIS), task_path="config/model/cost_list_model", layout_analysis=LAYOUT_ANALYSIS, device_id=-1),
] ]
for phhd in phhds: for phhd in phhds:
pk_phhd = phhd.pk_phhd pk_phhd = phhd.pk_phhd