python图片加水印 python图片批量加水印

转载

架构领航员 2023-10-06 18:23:26

文章标签 python图片加水印程序员编程 python Python 文章分类 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的图形文件，其效果如下图所示：

python图片加水印 python图片批量加水印_python

为图片添加图片水印

下面这一步，我们为图片添加图片水印。这一步需要借助的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()，指定图片的目录，水印图片的位置以及新图片保存的位置就可以了。运行代码，其效果如下动图所示：

python图片加水印 python图片批量加水印_程序员_02