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

Revert "尝试使用cpu进行识别"

Browse Source 
This reverts commit ab72f21eff.
This commit is contained in:
liuyebo
2024-07-01 11:14:22 +08:00
parent ab72f21eff
commit 85df6177b0
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
photo_review/photo_review.py
View File

@@ -395,12 +395,11 @@ 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, device_id=-1), task_path="config/model/settlement_list_model", layout_analysis=LAYOUT_ANALYSIS),
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, device_id=-1), task_path="config/model/cost_list_model", layout_analysis=LAYOUT_ANALYSIS),
] ]
for phhd in phhds: for phhd in phhds:
pk_phhd = phhd.pk_phhd pk_phhd = phhd.pk_phhd