yasirarism/MissKatyPyro
1
0
Fork 0
mirror of https://github.com/yasirarism/MissKatyPyro.git synced 2026-01-07 04:14:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Run async for cv2

Browse source
This commit is contained in:
Yasir Aris M 2024-07-29 10:33:08 +07:00 committed by GitHub
parent 9d3ee727ac
commit dae3a51b90
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 9 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
misskaty/helper/tools.py
View file

@ -7,6 +7,7 @@ import time
import cv2 import cv2
import numpy as np import numpy as np
from http.cookies import SimpleCookie from http.cookies import SimpleCookie
from pyrogram import utils
from re import match as re_match from re import match as re_match
from typing import Union from typing import Union
from urllib.parse import urlparse from urllib.parse import urlparse
@ -177,13 +178,13 @@ async def gen_trans_image(msg, path):
dl = await msg.download() dl = await msg.download()
# load image # load image
img = cv2.imread(dl) img = await utils.run_sync(cv2.imread(dl))
# convert to graky # convert to graky
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) gray = await utils.run_sync(cv2.cvtColor(img, cv2.COLOR_BGR2GRAY))
# threshold input image as mask # threshold input image as mask
mask = cv2.threshold(gray, 250, 255, cv2.THRESH_BINARY)[1] mask = await utils.run_sync(cv2.threshold(gray, 250, 255, cv2.THRESH_BINARY)[1])
# negate mask # negate mask
mask = 255 - mask mask = 255 - mask
@ -191,22 +192,22 @@ async def gen_trans_image(msg, path):
# apply morphology to remove isolated extraneous noise # apply morphology to remove isolated extraneous noise
# use borderconstant of black since foreground touches the edges # use borderconstant of black since foreground touches the edges
kernel = np.ones((3,3), np.uint8) kernel = np.ones((3,3), np.uint8)
mask = cv2.morphologyEx(mask, cv2.MORPH_OPEN, kernel) mask = await utils.run_sync(cv2.morphologyEx(mask, cv2.MORPH_OPEN, kernel))
mask = cv2.morphologyEx(mask, cv2.MORPH_CLOSE, kernel) mask = await utils.run_sync(cv2.morphologyEx(mask, cv2.MORPH_CLOSE, kernel))
# anti-alias the mask -- blur then stretch # anti-alias the mask -- blur then stretch
# blur alpha channel # blur alpha channel
mask = cv2.GaussianBlur(mask, (0,0), sigmaX=2, sigmaY=2, borderType = cv2.BORDER_DEFAULT) mask = await utils.run_sync(cv2.GaussianBlur(mask, (0,0), sigmaX=2, sigmaY=2, borderType = cv2.BORDER_DEFAULT))
# linear stretch so that 127.5 goes to 0, but 255 stays 255 # linear stretch so that 127.5 goes to 0, but 255 stays 255
mask = (2*(mask.astype(np.float32))-255.0).clip(0,255).astype(np.uint8) mask = (2*(mask.astype(np.float32))-255.0).clip(0,255).astype(np.uint8)
# put mask into alpha channel # put mask into alpha channel
result = img.copy() result = img.copy()
result = cv2.cvtColor(result, cv2.COLOR_BGR2BGRA) result = await utils.run_sync(cv2.cvtColor(result, cv2.COLOR_BGR2BGRA))
result[:, :, 3] = mask result[:, :, 3] = mask
# save resulting masked image # save resulting masked image
cv2.imwrite(path, result) await utils.run_sync(cv2.imwrite(path, result))
os.remove(dl) os.remove(dl)
return path return path