批量加水印防抄袭，Python轻松搞定！_图片加水印增加ocr难度 python-优快云博客

一、为什么要为图片添加水印

在互联网写文章最头疼的一件事情就是发表的文章容易被抄袭，各种抄袭的手段真的是层出不穷。无奈，只能在不影响阅读体验的情况下不断地提高反抄袭的难度。虽然防不住，但是为文章中的图片添加水印，至少能够保证文章被抄袭之后，读者还能知道文章的来源网站，但是一个一个加水印实在太麻烦，而且假的水印很容易被截掉。

索性，我们就用Python来完成一个图片水印批量添加程序的开发。

二、利用Python为图片添加水印

图片添加图片水印的原理其实很简单，在一个空白的画布上依次将原图像和水印添加添加进去即可。

在Python中，我们可以借助PIL模块轻松地实现图片水印的添加。

下面，我们通过代码来演示一下。

遍历文件夹下所有图片文件

首先我们需要获取一个目录下的所有文件，提取出其中的图像文件，这一步通过内置的os模块即可实现，代码如下所示：

import os
img_suffix_list = ['png', 'jpg', 'bmp']
for i in os.listdir(r"C:\Users\yang9\Desktop"):
    if i.split('.')[-1] in img_suffix_list:
        print(i)

运行上述代码，我们将获取选定目录下后缀名为png、jpg、bmp的图形文件，其效果如下图所示：

为图片添加图片水印

下面这一步，我们为图片添加图片水印。这一步需要借助的PIL的Image类来实现。

首先，读取原始图片和水印图片，并获取原始图片的大小，

然后我们新建一个图层，其大小为原始图片的大小，并将水印图片添加到图层上：

wm_position = (img_size[0]-wm_size[0],img_size[1]-wm_size[1]) # 默认设定水印位置为右下角
layer = Image.new('RGBA', img.size)  # 新建一个图层
layer.paste(watermark, wm_position)  # 将水印图片添加到图层上

接着，我们对图层和原始图片进行合成：

mark_img = Image.composite(layer, img, layer)

最后，调用save方法，将新图片保存到本地：

mark_img.save(save_path + new_file_name)

这样，一个添加了水印的图片就生成了。

我们将其整合为两个函数，方便调用，代码如下所示：

# coding:utf-8
import os,traceback
from PIL import Image

# 获取文件夹图片
def get_folder(fpath,wm_file,save_path):
    try:
        img_suffix_list = ['png', 'jpg', 'bmp']
        for i in os.listdir(fpath):
            if i.split('.')[-1] in img_suffix_list:
                img_path = fpath + '/' + i
                img_water_mark(img_file=img_path,wm_file=wm_file,save_path=save_path)
    except Exception as e:
        print(traceback.print_exc())

# 图片添加水印
def img_water_mark(img_file, wm_file,save_path):
    try:
        img = Image.open(img_file)  # 打开图片
        watermark = Image.open(wm_file)  # 打开水印
        img_size = img.size
        wm_size = watermark.size
        # 如果图片大小小于水印大小
        if img_size[0] < wm_size[0]:
            watermark.resize(tuple(map(lambda x: int(x * 0.5), watermark.size)))
        print('图片大小：', img_size)
        wm_position = (img_size[0]-wm_size[0],img_size[1]-wm_size[1]) # 默认设定水印位置为右下角
        layer = Image.new('RGBA', img.size)  # 新建一个图层
        layer.paste(watermark, wm_position)  # 将水印图片添加到图层上
        mark_img = Image.composite(layer, img, layer)
        new_file_name = '/new_'+img_file.split('/')[-1]
        mark_img.save(save_path + new_file_name)
    except Exception as e:
        print(traceback.print_exc())

这样，我们通过一个函数get_folder()，指定图片的目录，水印图片的位置以及新图片保存的位置就可以了。运行代码，其效果如下动图所示：