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-05 16:57:04 +08:00
parent fbd6834dd5
commit 5498d62542
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
photo_review/photo_review.py
View File

@@ -28,7 +28,8 @@ def information_extraction(schema, phrecs, task_path):
for phrec in phrecs: for phrec in phrecs:
pic_path = get_private_url(phrec.cfjaddress) pic_path = get_private_url(phrec.cfjaddress)
if pic_path: if pic_path:
ie = Taskflow("information_extraction", schema=schema, model="uie-x-base", task_path=task_path) ie = Taskflow("information_extraction", schema=schema, model="uie-x-base", task_path=task_path,
layout_analysis=True)
# 批量抽取写法：(ie([{"doc": "./data/6.jpg"}, {"doc": "./data/7.jpg"}]) # 批量抽取写法：(ie([{"doc": "./data/6.jpg"}, {"doc": "./data/7.jpg"}])
result = ie({"doc": pic_path}) result = ie({"doc": pic_path})