Python 水印 - 防止图像尺寸和质量降低

如何解决Python 水印 - 防止图像尺寸和质量降低

我有这个脚本，它成功地将水印应用于图像，调整水印大小并设置图像“质量”。 我发现它实际上降低了质量并将其调整为比我想要的小。

这是我正在使用的电报机器人，我将图像发送给机器人，机器人正在添加水印，然后调整大小。我根本不想要调整大小的部分。我要原尺寸+水印。

问题：如何更改此设置，使其不会调整添加水印的原始图像的大小？

电流

1. 我将图像发送到我的机器人以激活脚本
2. 脚本添加水印并调整图像大小
3. 脚本将调整后的图像连同水印一起发送给我

期望的流量

1. 我将图像发送到我的机器人以激活脚本
2. 脚本添加了水印但不会调整原始图像的大小
3. 脚本将包含水印的图像发送回给我

代码 1

# -*- coding: utf-8 -*-
import telebot
import config
import random
import os
import time
from tools.tools import getting_ready

bot = telebot.TeleBot(config.BOT_TOKEN)


@bot.message_handler(commands=['start','help'])
def send_welcome(message):
    bot.send_message(message.chat.id,'Hi,Send or Forward me a PHOTO!')


@bot.message_handler(func=lambda message: True)
def echo_all(message):
    bot.reply_to(message,'Send /start or /help')


@bot.message_handler(content_types=["photo"])
def send_watermark(message):
    chat_id = message.chat.id
    sent_message = bot.reply_to(message,'Downloading...')
    message_id = sent_message.message_id
    file = bot.get_file(message.photo[0].file_id)
    downloaded_file = bot.download_file(file.file_path)
    bot.edit_message_text(f'Downloaded!\n\nNow,Genarating Watermarks...\n\ncurrent watermark: {config.WATERMARK}',chat_id=chat_id,message_id=message_id)
    __path = 'images/' + str(random.randint(100000,999999)) + '.tmp'
    with open(__path,'wb') as f:
        f.write(downloaded_file)
    fname = getting_ready(__path)
    os.remove(__path)
    time.sleep(2)
    bot.edit_message_text('Now,Uploading...',message_id=message_id)
    for i in ('black','white',):
        __file = 'images/out/{}/{}'.format(i,fname)
        __photo = open(__file,'rb')
        bot.send_photo(message.chat.id,__photo,reply_to_message_id=message.message_id)
        os.remove(__file)
    bot.delete_message(chat_id=chat_id,message_id=message_id)


print('watermark bot started successfully!')
bot.polling()

代码 2

from PIL import Image,ImageFont,ImageDraw,ImageEnhance
import hashlib
import piexif
import config


def getting_ready(path):
    fname = md5(path) + '.jpg'
    image = Image.open(path).convert('RGB')
    try:
        exif_dict = piexif.load(image.info['exif'])
        orientation = exif_dict['0th'][274]
    except KeyError:
        orientation = None
    rotate_values = {3: 180,6: 270,8: 90}
    if orientation in rotate_values:
        image = image.rotate(rotate_values[orientation],expand=True)
    for color in ['black','white']:
        watermark(image,fname,color)
    return fname


def watermark(img,new_fname,color):
    text = config.WATERMARK
    wm = Image.new('RGBA',img.size,(0,0))
    fontsize = img.size[1] // 100 * config.FONT_SIZE
    font = ImageFont.truetype(f'tools/fonts/{config.FONT_NAME}',fontsize)
    indent = fontsize // 8
    w,h = font.getsize(text)
    text_position = (img.size[0] - w - indent,img.size[1] - h - indent)
    draw = ImageDraw.Draw(wm,'RGBA')
    draw.text(text_position,text,font=font,fill=color)
    alpha = wm.split()[3]
    alpha = ImageEnhance.Brightness(alpha).enhance(config.TRANSPARENCY)
    wm.putalpha(alpha)
    out_path = 'images/out/{}/{}'.format(color,new_fname)
    Image.composite(wm,img,wm).save(out_path,'JPEG',optimize=False,quality=config.QUALITY)


def md5(path):
    with open(path,'rb') as f:
        md5hash = hashlib.md5()
        for chunk in iter(lambda: f.read(4096),b''):
            md5hash.update(chunk)
    return md5hash.hexdigest()

解决方法

这可能不是解决此问题的最佳方法，但它对我来说很有效，可以将其恢复到正常大小

double = 2.0
    Image.composite(wm,img,wm).resize( [int(double * s) for s in img.size] ).save(out_path,'JPEG',optimize=True,quality=config.QUALITY)