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: