亲宝软件园·资讯

展开

python水印生成器

东华果汁哥 人气:0

制作一个水印生成器,方便训练水印图片数据采集。
图片水印生成器,可以给指定图片文件或者目录添加水印,水印支持自定义文本、位置、颜色、大小。

# -*- encoding=utf-8 -*-
import time
from PIL import Image, ImageDraw,ImageFont,ImageEnhance

"""图片水印生成器,自定义文字,颜色,大小,位置"""


# 横向上方水印
def auto_make_watermark1(filepath,content,color,savefilepath):
    """
    :param filepath: 图片路径
    :param content: 水印文字
    :param color: 水印颜色
    :param savefilepath: 保存路径
    :return:
    """
    image = Image.open(filepath).convert('RGB')
    draw = ImageDraw.Draw(image)
    font=ImageFont.truetype("simsun.ttc", 40, encoding="unic",index=1)  # 设置水印字体
    draw.text((80, 200), content,color, font) # 设置水印位置
    image.save(savefilepath)

# 横向中间水印
def auto_make_watermark2(filepath,content,color,savefilepath):
    """
    :param filepath: 图片路径
    :param content: 水印文字
    :param color: 水印颜色
    :param savefilepath: 保存路径
    :return:
    """
    image = Image.open(filepath).convert('RGB')
    draw = ImageDraw.Draw(image)
    font=ImageFont.truetype("simsun.ttc", 40, encoding="unic",index=1)  # 设置水印字体
    draw.text((80, 400), content,color, font)   # 设置水印位置
    image.save(savefilepath)

# 横向最下方水印
def auto_make_watermark3(filepath,content,color,savefilepath):
    """
    :param filepath: 图片路径
    :param content: 水印文字
    :param color: 水印颜色
    :param savefilepath: 保存路径
    :return:
    """
    image = Image.open(filepath).convert('RGB')
    draw = ImageDraw.Draw(image)
    font=ImageFont.truetype("simsun.ttc", 40, encoding="unic",index=1)  # 设置字体
    draw.text((80,image.size[1]-150), content,color, font) # 设置水印位置
    image.save(savefilepath)

# 横向中间倾斜水印45度
def auto_make_watermark4(filepath,content,color,savefilepath,radio):
    """
    :param filepath: 图片路径
    :param content: 水印文字
    :param color: 水印颜色
    :param savefilepath: 保存路径
    :param radio: 水印倾斜角度
    :return:
    """
    im = Image.open(filepath)
    watermark = Image.new('RGBA', im.size)
    draw = ImageDraw.Draw(watermark, 'RGBA')
    font = ImageFont.truetype("simsun.ttc", 40, encoding="unic", index=1)
    # x y 坐标
    draw.text((80, 400), content, font=font,  fill=color)
    # 旋转45度
    watermark = watermark.rotate(radio, Image.BICUBIC)
    # 透明的
    alpha = watermark.split()[3]
    alpha = ImageEnhance.Brightness(alpha).enhance(0.7)
    watermark.putalpha(alpha)
    # 合成新的图片
    image2 = Image.composite(watermark, im, watermark)
    image2.save(savefilepath)

if __name__ == '__main__':
    time1=time.time()
    filepath=r'F:/img_spam/python添加水印/10064003738101263000320010013284.jpg'

    savefilepath='F:/img_spam/python添加水印/生成/'
    text="诚招发单人,vx:18668124728"
    colors=['red','green','black']

    for color in colors:
        auto_make_watermark1(filepath,text,color,savefilepath+'10064003738101263000320010013285'+'_'+color+'_'+'top'+'.jpg')
        auto_make_watermark2(filepath, text, color,savefilepath + '10064003738101263000320010013285' + '_' + color+'_' +'mmddle'+'.jpg')
        auto_make_watermark3(filepath, text, color,savefilepath + '10064003738101263000320010013285' + '_' + color +'_'+ 'down' + '.jpg')
        auto_make_watermark4(filepath, text, color,savefilepath + '10064003738101263000320010013285' + '_' + color+'_' + 'angle' + '.png',45)
        auto_make_watermark4(filepath, text, color,savefilepath + '10064003738101263000320010013285' + '_' + color + '_' +'angle_50' + '.png', -50)

    time2=time.time()
    print('总共耗时:' + str(time2 - time1) + 's')

加载全部内容

相关教程
猜你喜欢
用户评论