python批量添加水印_Python实现批量图片添加数字水印
之前写过一个类似的代码，是把水印信息打散以后随机添加到原图中，并提供了水印信息的提取功能，请参考Python实现图像空域随机水印加入与提取。本文代码功能：为指定文件夹中的所有图像文件批量添加水印，水印位置在左上角、中间、右下角这三个位置中随机选择。
from random import randint
from os import listdir
from PIL import Image
#打开并读取其中的水印像素，也就是不是白色背景的像素
#读到内存中，放到字典中以供快速访问
im = Image.open('watermark.bmp')
width, height = im.size
pixels = dict()
for w in range(width):
for h in range(height):
c = im.getpixel((w,h))[:3]
if c!=(255, 255, 255):
pixels[(w, h)] = c
def addWaterMark(srcDir):
#获取当前所有BMP图像文件列表
picFiles = [fn for fn in listdir(srcDir) if fn.endswith(('.bmp', '.jpg', '.png'))]
#遍历所有文件，为每个图像添加水印
for fn in picFiles:
im1 = Image.open(fn)
w, h = im1.size
#如果图片尺寸小于水印图片，不加水印
if w
continue
#在原始图像左上角、中间或右下角添加数字水印
#具体位置根据position进行随机选择
p = {0:(0,0), 1:((w-width)//2, (h-height)//2), 2:(w-width, h-height)}
position = randint(0,2)
top, left = p.get(position, (0,0))
#修改像素值，添加水印
for p, c in pixels.items():
im1.putpixel((p[0]+top, p[1]+left), c)
#保存加入水印之后的新图像文件
im1.save(fn[:-4] + '_new' + fn[-4:])
#为当前文件夹中的图像文件添加水印
addWaterMark('.')
其中一张效果图：
