图片批量添加文字水印

为了版权安全,可以给自己设计的图片设置批量文字水印功能

思路分析

为图片添加文字水印,主要用到PIL库中的相应类,PIL是一个Python图像处理库,它支持多种文件格式,并提供了强大的图像处理和图形处理能力,本例中主要使用PIL库的Image模块、ImageDraw模块、ImageFont模块和ImageEnhance模块,作用分别如下:

  • Image模块:该模块中提供了一个与模块同名的Image类,用来对图像进行处理,另外,它还提供从文件加载图像和创建新图像的功能。
  • ImageDraw模块:该模块提供简单的2D图形对象,在本例中,使用该模块中提供的draw()方法来绘制水印文字。
  • ImageFont模块:该模块中提供了一个与模块同名的ImageFont类,主要用来存储字体,它通常与ImageDraw.Draw.text()方法一起使用。
  • ImageEnhance模块:该模块中包含了一些用于图像增强的类,本例中设置水印文字的透明度时用到该模块中Brightness类的enhance()方法

初始图片

金牛拜年.png

牛气冲天.png

首先对水印相关的内容进行设置,并使用Image对象的open()方法和new()方法分别打开原始图片和新建一个存储水印图片的对象;然后使用ImageDraw.Draw对象的text()方法绘制水印文字,通过ImageEnhance模块中Brightness类的enhance()方法设置水印透明度;最后使用save()方法将图像保存为png格式。

代码参考

# -*- coding: utf-8 -*-
"""
@Time : 2021/2/13 18:31
@Auth : Yvon~早安阳光
@File :shuiyin
"""
from PIL import Image, ImageDraw, ImageFont,ImageEnhance
import os
path=input('请输入要添加水印的图片所在路径:')
text=input('请输入水印文字:')
positionflag=int(input('请输入水印位置(0:左上角,1:左下角,2:右上角,3:右下角,4:居中):'))
alphavalue=float(input('请输入水印透明度(范围在0――1之间的1位小数):'))

# 设置所使用的字体
font = ImageFont.truetype(r'C:\Windows\Fonts\SIMLI.TTF', 80)
# 文字水印
def textMark(img):
    try:
        im = Image.open(img).convert('RGBA')             # 打开原始图片,并转换为RGBA色值
        newImg = Image.new('RGBA', im.size, (255, 255, 255, 0)) # 存储添加水印后的图片
        imagedraw = ImageDraw.Draw(newImg)               # 创建绘制对象
        imgwidth, imgheight = im.size                    # 记录图片大小
        txtwidth=font.getsize(text)[0]                   # 获取字体宽度
        txtheight = font.getsize(text)[1]                # 获取字体高度

        # 设置水印文字位置
        if positionflag == 0:                             # 左上角
            position=(0,0)
        elif  positionflag == 1:                          # 左下角
             position=(0,imgheight - txtheight)
        elif  positionflag == 2:                          # 右上角
            position=(imgwidth - txtwidth,0)
        elif  positionflag == 3:                          # 右下角
            position=(imgwidth - txtwidth, imgheight - txtheight)
        elif  positionflag == 4:                          # 居中
            position=(imgwidth/2,imgheight/2)

        # 绘制文字
        imagedraw.text(position, text, font=font, fill="yellow")
        # 设置透明度
        alpha = newImg.split()[3]
        alpha = ImageEnhance.Brightness(alpha).enhance(alphavalue)
        newImg.putalpha(alpha)
        Image.alpha_composite(im, newImg).save(img,'png')# 保存图片
    except Exception as e:
        print(e)
try:
    list = os.listdir(path)                             # 遍历选择的文件夹
    for i in range(0, len(list)):                      # 遍历文件列表
        filepath = os.path.join(path, list[i])        # 记录遍历到的文件名
        if os.path.isfile(filepath):                   # 判断是否为文件
            filetype = os.path.splitext(filepath)[1]  # 获取扩展名
            if filetype == '.png':                      # 判断是否为.png
                textMark(filepath)                      # 批量添加文字水印
    print('批量添加水印完成')
except:
    print('请输入一个有效路径……')

运行结果

输入参数.png

金牛拜年.png

牛气冲天.png

你可能感兴趣的:(图片批量添加文字水印)