欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

使用python PIL 模块合成图片

程序员文章站 2022-07-10 14:33:27
...

使用python PIL 模块合成图片

需求背景:

公司产品有一套完整的用户手册,但是给用户部署时大多数为私有化部署。给用户用户手册时,我们本来的产品logo需要更换为客户的logo。重新截图或者手动更改需要大量时间。

脚本实现:

该脚本实现对用户手册中的图片进行合成,更换截图中的logo为客户的logo。从而节省修改用户手册时间。

代码块

from PIL import Image
import os

path_dir = '/Users/dtwave/Desktop/a/'   #图片1下的路径

path_save = '/Users/dtwave/Desktop/b/'

path_2 = "/Users/dtwave/Desktop/aaa.png"

def blend_two_images2(path_dir,path_2,path_save):
    dir = os.listdir(path_dir)                  # dir是目录下的全部文件
    print (dir)
    for d in dir:
        # a.append(d)
        # for i in range(2,17):
        img1 = Image.open(path_dir + d)    #打开图片1
        img1 = img1.convert('RGBA')

        img2 = Image.open(path_2)   #打开图片2
        img2 = img2.convert('RGBA')

        r, g, b, alpha = img2.split()   #设置图片2透明度
        alpha = alpha.point(lambda i: i>0 and 1000)

        img = Image.composite(img2, img1, alpha)   #开始合成

        img.show()        #是否打印出合成后的图片
        img.save(path_save + d)   #保存在在一个文件目录下

    return
a = blend_two_images2(path_dir,path_2,path_save)

脚本写的相对粗糙,有比较多的局限,例如图片上logo大小与合成图片大小不符,就导致合成盖住的地方多了或者少了。后续会优化脚本,利用python的图像识别,自动抠图,与合成图片对比大小。将合成图片缩放至同样大小再进行合成