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

修正ocr没有识别到内容导致的空指针错误

Browse Source
This commit is contained in:
liuyebo
2024-10-17 16:59:10 +08:00
parent 39da0d8a00
commit cc53243647
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
photo_review/auto_photo_review.py
View File

@@ -141,7 +141,9 @@ def information_extraction(phrec, pk_phhd):
if split_result['img'] is None: if split_result['img'] is None:
continue continue
a4_img = image_util.expand_to_a4_size(split_result['img']) a4_img = image_util.expand_to_a4_size(split_result['img'])
ocr_result += model_util.ocr(a4_img) tmp_ocr_result = model_util.ocr(a4_img)
if tmp_ocr_result:
ocr_result += tmp_ocr_result
tmp_ocr_text = common_util.ocr_result_to_text(ocr_result) tmp_ocr_text = common_util.ocr_result_to_text(ocr_result)
tmp_rec_type = model_util.clas_text(tmp_ocr_text) if ocr_text else None tmp_rec_type = model_util.clas_text(tmp_ocr_text) if ocr_text else None