jarvis/jarvis-bot
jarvis/jarvis-bot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Very poor target meeting

Browse source
This commit is contained in:
Zeva Rose 2021-06-26 20:46:32 -06:00
parent 9a35051a71
commit d5d23c6f2c
1 changed files with 11 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
jarvis/cogs/image.py
View file

@ -58,15 +58,19 @@ class ImageCog(commands.Cog):
ratio = tgt_size / size
buffer = np.frombuffer(file, dtype=np.uint8)
img = cv2.imdecode(buffer, flags=-1)
# TODO: Optimize to not run multiple times
while len(file) > tgt_size:
width = int(img.shape[1] * ratio)
height = int(img.shape[0] * ratio)
buffer = np.frombuffer(file, dtype=np.uint8)
img = cv2.imdecode(buffer, flags=-1)
new_img = cv2.resize(img, (width, height))
buffer = cv2.imencode(".png", new_img)[1].tobytes()
bufio = BytesIO(buffer)
width = int(img.shape[1] * ratio)
height = int(img.shape[0] * ratio)
new_img = cv2.resize(img, (width, height))
file = cv2.imencode(".png", new_img)[1].tobytes()
bufio = BytesIO(file)
await ctx.send(file=File(bufio, filename="resized.png"))