From 9bd9c9d7054b412d10a6d20552eae031eadb7347 Mon Sep 17 00:00:00 2001 From: liuyebo <1515783401@qq.com> Date: Mon, 1 Jul 2024 09:46:31 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B2=A1=E6=9C=89=E8=AF=86=E5=88=AB=E7=BB=93?= =?UTF-8?q?=E6=9E=9C=E4=B9=9F=E4=BF=9D=E5=AD=98=E5=9B=BE=E7=89=87?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- check_ie_result/check_ie_result.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/check_ie_result/check_ie_result.py b/check_ie_result/check_ie_result.py index a072726..6a63a31 100644 --- a/check_ie_result/check_ie_result.py +++ b/check_ie_result/check_ie_result.py @@ -78,6 +78,10 @@ def check_ie_result(pk_phhd): font = ImageFont.truetype("./font/simfang.ttf", size=size) ocr = session.query(ZxOcr.id, ZxOcr.content, ZxOcr.x_offset, ZxOcr.y_offset).filter(ZxOcr.pk_phrec == phrec.pk_phrec).all() + if not ocr: + os.makedirs(f"./check_result/{pk_phhd}/0", exist_ok=True) + image.save(f"./check_result/{pk_phhd}/0/{img_name}") + for id, group_results in groupby(ocr, key=lambda x: x.id): draw = ImageDraw.Draw(image) for ocr_item in group_results: