快捷導(dǎo)航

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

更新時(shí)間：2019年11月30日 09:25:28 作者：白玉無(wú)冰

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

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

配置環(huán)境

python3 + pillow

pip3 install pillow

引入庫(kù)

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

混合顏色：

對(duì)每一個(gè)像素點(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 對(duì)象：

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

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;

處理單個(gè)文件：

對(duì)于 gif 文件先拆成一張一張圖片，在圖片上添加水印，最后再合成 gif 。對(duì)于其他格式的圖片文件可以多添加幾個(gè)水印。最后輸出保存到 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}')

處理目錄：

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

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 庫(kù)來(lái)實(shí)現(xiàn)。首先是讀取 logo 圖片信息，接著在隨機(jī)一個(gè)位置添加混合后的像素點(diǎn)信息，最后再保存起來(lái)。

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

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

您可能感興趣的文章: