Python實(shí)現(xiàn)圖片批量加入水印代碼實(shí)例

更新時間：2019年11月30日 09:25:28 作者：白玉無冰

這篇文章主要介紹了Python實(shí)現(xiàn)圖片批量加入水印代碼實(shí)例,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下

寫文章的時候可以設(shè)置是否添加水印?？墒?，有些圖片可能想加水印，有些不想加水印，該怎么辦呢？

配置環(huán)境

python3 + pillow

pip3 install pillow

引入庫

from PIL import Image, ImageSequence
import os
import random

效果預(yù)覽：

使用方法：

在腳本同目錄下添加水印圖片 logo.png
創(chuàng)建目錄 input 并在放入要添加水印的圖片
創(chuàng)建目錄 output 執(zhí)行腳本 addlogo.py
結(jié)果輸出在 output 文件夾下

實(shí)現(xiàn)原理

水印圖片采集：

先讀取水印圖片的像素點(diǎn)信息和大小信息。去除透明度為0的像素，并修改透明度像素信息。

img_logo = Image.open("logo.png")
img_logo_width, img_logo_height = img_logo.size
img_logo_pixels = dict()
for w in range(img_logo_width):
  for h in range(img_logo_height):
    c = img_logo.getpixel((w,h))
    if c!=(0, 0, 0, 0):
      img_logo_pixels[(w, h)] = c[:3]+(125,)

混合顏色：

對每一個像素點(diǎn)采取顏色混合，其中c1是源圖片的像素點(diǎn)信息(r,g,b,a)，c2是logo圖片像素點(diǎn)的信息。混合算法如下:

def blendPixel(c1,c2):
  a1=256-c2[3]
  a2=c2[3]-(a1*c2[3])/256.0
  a=a1+a2
  c=(int((a1*c1[0] + a2*c2[0])/a), int((a1*c1[1] + a2*c2[1])/a), int((a1*c1[2] + a2*c2[2])/a),int(a))
  return c

處理源 Image 對象：

隨機(jī)一個位置開始處理像素，具體代碼參考如下。

def dealOneImage(image,offX=None,offY=None):
  w, h = image.size
  offX = offX if offX else random.random();
  offY = offY if offY else random.random();
  #如果圖片尺寸小于水印圖片，不加水印
  if w>=img_logo_width and h>=img_logo_height:
    top = int((w - img_logo_width)*offX)
    left = int((h - img_logo_height)*offY)
    for p, c in img_logo_pixels.items():
      p_x = (p[0]+top)
      p_y = (p[1]+left)
      image_c = image.getpixel((p_x,p_y))
      if(isinstance(image_c, tuple) and len(image_c)>2):
        image.putpixel((p_x, p_y), blendPixel(image_c,c))
  return image;

處理單個文件：

對于 gif 文件先拆成一張一張圖片，在圖片上添加水印，最后再合成 gif 。對于其他格式的圖片文件可以多添加幾個水印。最后輸出保存到 output 文件夾下。

def dealOneFile(filePath):
  img_orign = Image.open(filePath)
  _,file_type = os.path.splitext(filePath)
  basename = os.path.basename(filePath)
  if file_type == '.gif':
    sequence = [];
    offX=random.random()
    offY=random.random()
    for f in ImageSequence.Iterator(img_orign):
      if len(sequence) % 2 == 0:
        offX=random.random()
        offY=random.random()
      sequence.append(dealOneImage(f.convert(),offX,offY))
    sequence[0].save(f'./output/{basename}', save_all=True, append_images=sequence[1:])
  else:
    image_out = (dealOneImage(img_orign))
    for x in range(1):
      image_out = (dealOneImage(image_out))
    image_out.save(f'./output/{basename}')

處理目錄：

對當(dāng)前目錄下的文件做一個篩選，只選取圖片格式的文件。

def dealSrc(srcDir):
  picFiles = [os.path.join(srcDir,fn) for fn in os.listdir(srcDir) if fn.endswith(('.gif', '.jpg', '.png','.jpeg'))]
  for filePath in picFiles:
    dealOneFile(filePath)

小結(jié)

添加水印主要用了 python3 中的 pillow 庫來實(shí)現(xiàn)。首先是讀取 logo 圖片信息，接著在隨機(jī)一個位置添加混合后的像素點(diǎn)信息，最后再保存起來。

其中，對于 gif 文件的處理是拆幀，再添加水印，最后再組合成一個 gif。這個只能對應(yīng)比較小的 gif 文件處理，如果有更好的方法歡迎留言交流分享！

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: