Update img_utils.py

This commit is contained in:
HibiKier 2021-07-27 17:27:31 +08:00 committed by GitHub
parent 0dfca7e23d
commit edbbcec1be
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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)