yasirarism/MissKatyPyro
1
0
Fork 0
mirror of https://github.com/yasirarism/MissKatyPyro.git synced 2025-12-29 09:44:50 +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 numpy as np
from http.cookies import SimpleCookie
from pyrogram import utils
from re import match as re_match
from typing import Union
from urllib.parse import urlparse
@ -177,13 +178,13 @@ async def gen_trans_image(msg, path):
dl = await msg.download()
# load image
img = cv2.imread(dl)
img = await utils.run_sync(cv2.imread(dl))
# 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
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
mask = 255 - mask
@ -191,22 +192,22 @@ async def gen_trans_image(msg, path):
# apply morphology to remove isolated extraneous noise
# use borderconstant of black since foreground touches the edges
kernel = np.ones((3,3), np.uint8)
mask = 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_OPEN, kernel))
mask = await utils.run_sync(cv2.morphologyEx(mask, cv2.MORPH_CLOSE, kernel))
# anti-alias the mask -- blur then stretch
# 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
mask = (2*(mask.astype(np.float32))-255.0).clip(0,255).astype(np.uint8)
# put mask into alpha channel
result = img.copy()
result = cv2.cvtColor(result, cv2.COLOR_BGR2BGRA)
result = await utils.run_sync(cv2.cvtColor(result, cv2.COLOR_BGR2BGRA))
result[:, :, 3] = mask
# save resulting masked image
cv2.imwrite(path, result)
await utils.run_sync(cv2.imwrite(path, result))
os.remove(dl)
return path