From edbbcec1be4f5611e829b187819e96f0decffc93 Mon Sep 17 00:00:00 2001 From: HibiKier <45528451+HibiKier@users.noreply.github.com> Date: Tue, 27 Jul 2021 17:27:31 +0800 Subject: [PATCH] Update img_utils.py --- utils/img_utils.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/utils/img_utils.py b/utils/img_utils.py index d1491408..3bbb2293 100644 --- a/utils/img_utils.py +++ b/utils/img_utils.py @@ -2,6 +2,7 @@ import os from configs.path_config import IMAGE_PATH, TXT_PATH, TTF_PATH from PIL import Image, ImageFile, ImageDraw, ImageFont import cv2 +from typing import Tuple import imagehash import base64 from io import BytesIO @@ -223,6 +224,18 @@ class CreateImg: self.current_w += self.img_w return self.markImg + def crop(self, box: Tuple[int, int, int, int]): + """ + 说明: + 裁剪图片 + 参数: + :param box: 左上角坐标,右下角坐标 (left, upper, right, lower) + """ + self.markImg = self.markImg.crop(box) + self.w, self.h = self.markImg.size + self.size = self.w, self.h + self.draw = ImageDraw.Draw(self.markImg) + # 获取文字大小 def getsize(self, msg): return self.ttfont.getsize(msg)