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-10-15 14:21:23 +08:00
parent b09f16fe23
commit 445d57e8c6
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
photo_review/auto_photo_review.py
View File

@@ -140,6 +140,7 @@ def information_extraction(phrec, pk_phhd):
dewarped_img_path = img_path dewarped_img_path = img_path
angles = model_util.clas_orientation(dewarped_img_path) angles = model_util.clas_orientation(dewarped_img_path)
ocr_result = [] ocr_result = []
ocr_text = ''
rotated_img = None rotated_img = None
for angle in angles: for angle in angles:
tmp_ocr_result = [] tmp_ocr_result = []
@@ -150,10 +151,12 @@ def information_extraction(phrec, pk_phhd):
continue continue
a4_img = image_util.expand_to_a4_size(split_result['img']) a4_img = image_util.expand_to_a4_size(split_result['img'])
tmp_ocr_result += model_util.ocr(a4_img) tmp_ocr_result += model_util.ocr(a4_img)
if len(tmp_ocr_result) > len(ocr_result): tmp_ocr_text = common_util.ocr_result_to_text(tmp_ocr_result)
if len(tmp_ocr_text) > len(ocr_text):
ocr_result = tmp_ocr_result ocr_result = tmp_ocr_result
ocr_text = tmp_ocr_text
rotated_img = tmp_rotated_img rotated_img = tmp_rotated_img
ocr_text = common_util.ocr_result_to_text(ocr_result)
rec_type = model_util.clas_text(ocr_text) if ocr_text else None rec_type = model_util.clas_text(ocr_text) if ocr_text else None
if rec_type == '基本医保结算单': if rec_type == '基本医保结算单':
info_extract = model_util.ie_settlement(rotated_img, common_util.ocr_result_to_layout(ocr_result)) info_extract = model_util.ie_settlement(rotated_img, common_util.ocr_result_to_layout(ocr_result))