技术干货！用Python批量为图片添加文字水印

我们平时在天猫购买商品时，每种商品的宣传图片中都会有透明度不同的商家店铺名称，我们把这种文字叫做水印文字，效果类似于图1，这样做的好处是，可以防止其他店铺直接盗用自己的图片去使用，那么如何通过Python实现为图片批量添加水印文字呢？本实例将实现批量为图片添加文字水印。                     

图1  日常看到的添加水印的内容

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

• Image模块 该模块中提供了一个与模块同名的Image类，用来对图像进行处理，另外，它还提供从文件加载图像和创建新图像的功能。

• ImageDraw模块 该模块提供简单的2D图形对象，在本例中，使用该模块中提供的draw()方法来绘制水印文字。

• ImageFont模块 该模块中提供了一个与模块同名的ImageFont类，主要用来存储字体，它通常与ImageDraw.Draw.text()方法一起使用。

• ImageEnhance模块 该模块中包含了一些用于图像增强的类，本例中设置水印文字的透明度时用到该模块中Brightness类的enhance()方法。

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

from PIL import Image, ImageDraw, ImageFont,ImageEnhance

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', 24)

# 文字水印

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="red")

        # 设置透明度

        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)

import os

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('请输入一个有效路径……')

说明 添加水印的图片不能是jpg 者jpeg格式，因为jpg 者jpeg格式的图片不支持透明度的设置。

运行程序，输入需要添加水印的图片所在路径和要添加的水印文字，设置水印位置和透明度，如图2所示，按<Enter>回车键，即可批量为指定路径下的所有图片添加文字水印，效果如图3所示。

图2  设置文字水印的相关信息

图3  添加文字水印后的图片效果

更多有价值资源尽在程序开发资源库。

点击“ ”，前往程序开发资源库