现在打算将一篇店铺素材批量加logo后上传至电商平台,平台要求上传像素为800*800。
下面我们来一步步使用Python来实现:
测试
首先读取logo图片看看效果:
from PIL import Image
logo = Image.open('logo2.png')
logo
读取后发现logo图片存在大量空白:
我们可以人工编辑图片删除空白,也可以通过程序来自动裁切。
测试一下裁切功能:
可以看到已经完美去掉的四周的空白。
任意读一张主图:
按照像素为800*800的要求,我们需要先对图片的大小进行调整。可以直接拉伸也可以先裁切再拉伸。
这是直接拉伸的效果:
或者先将图片裁切到长宽一致后,再调整像素为800*800:
可以看到两种处理方式各有千秋,我们可以编写成可以传参的函数,供用户使用。
整理最终代码如下:
批量加logo的完整代码
from PIL import Image, ImageChops
from pathlib import Path
import os
def crop_image(im):
'''裁剪图片边缘空白'''
bg = Image.new(mode='RGBA', size=im.size)
bbox = ImageChops.difference(im, bg).getbbox()
if bbox:
return im.crop(bbox)
return im
def img_append_logo(logo, src, dest, size=(800, 800), crop=True):
logo = crop_image(Image.open(logo))
l_w, l_h = logo.size
if not os.path.exists(dest):
os.makedirs(dest)
for file in Path(src).glob("*.jpg"):
im = Image.open(file)
if crop:
w, h = im.size
d = abs(w-h)/2
if w >= h:
im = im.crop((d, 0, w-d, h))
else:
im = im.crop((0, d, 0, h-d))
im = im.resize(size)
w_r = size[0]//4
new_h = l_h*w_r//l_w
im.paste(logo.resize((w_r, new_h)), (0, 0))
im.save(f"{dest}/{file.name}")
img_append_logo('logo2.png', "img_src", "result")
不裁切再运行一次看看效果:
img_append_logo('logo2.png', "img_src", "result2", crop=False)