From e7105eb2e4132d74087a3fcbbc3a0f4a36fc138a Mon Sep 17 00:00:00 2001 From: liuyebo <1515783401@qq.com> Date: Thu, 20 Jun 2024 17:28:10 +0800 Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4zx=5Focr=E7=9A=84=E5=AD=98?= =?UTF-8?q?=E8=A1=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- photo_review/photo_review.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/photo_review/photo_review.py b/photo_review/photo_review.py index 5a725f9..fd95064 100644 --- a/photo_review/photo_review.py +++ b/photo_review/photo_review.py @@ -4,6 +4,7 @@ import math import os import sys import tempfile +import time from io import BytesIO import paddle @@ -99,7 +100,7 @@ def information_extraction(ie, phrecs): with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as temp_file: img["img"].save(temp_file.name) docs.append({"doc": temp_file.name}) - doc_phrecs.append(phrec) + doc_phrecs.append({"phrec": phrec, "x_offset": img["x_offset"], "y_offset": img["y_offset"]}) if not docs: return result @@ -117,15 +118,18 @@ def information_extraction(ie, phrecs): logging.info(f"删除临时文件 {temp_file['doc']} 时出错: {e}") now = get_default_datetime() + id = int(time.time()) for i in range(len(ie_results)): ie_result = ie_results[i] - phrec = doc_phrecs[i] + doc_phrec = doc_phrecs[i] + phrec = doc_phrec["phrec"] result_json = json.dumps(ie_result, ensure_ascii=False) if len(result_json) > 5000: result_json = result_json[:5000] session = MysqlSession() - zx_ocr = ZxOcr(pk_phhd=phrec.pk_phhd, pk_phrec=phrec.pk_phrec, cfjaddress=phrec.cfjaddress, - content=result_json, create_time=now, update_time=now) + zx_ocr = ZxOcr(pk_phhd=phrec.pk_phhd, pk_phrec=phrec.pk_phrec, id=id, cfjaddress=phrec.cfjaddress, + content=result_json, x_offset=doc_phrec["x_offset"], y_offset=doc_phrec["y_offset"], + create_time=now, update_time=now) session.add(zx_ocr) session.commit() session.close()