「码点代码」Python 实现图片拼接

Python 实现图片拼接

最近写了一片文章:「有点好奇」我都追过哪些美剧,文章配图时想把看过的美剧海报拼接成一张图。

第一个想到的是使用 PS 导入拼接,但这种做法显得不够极客,于是决定使用 Python 代码实现图片拼接。

首先原图片是 24 张美剧海报图片,需要将其拼接为一张 3*8 的图片。

上图拼接效果如下:

代码实现

对于图片的合并,需要读取本地图片,同时使用到 PIL 模块。因此如果你未安装此模块,可以使用 pip 进行安装。

模块代码如下:

1
2
import PIL.Image as Image
import os

随后定义一些常量:

1
2
3
4
5
6
7
IMAGES_PATH = './serials/'  # 图片集地址
IMAGE_WIDTH = 200 # 每张小图片的宽度
IMAGE_HEIGHT = 300 # 每张小片的高度
IMAGE_ROW = 3 # 图片间隔,也就是合并成一张图后,一共有几行
IMAGE_COLUMN = 8 # 图片间隔,也就是合并成一张图后,一共有几列
IMAGES_FORMAT = ['.jpg', '.JPG', 'png'] # 图片格式
IMAGE_SAVE_PATH = 'final.jpg' # 图片转换后的地址

这里需要对读取到照片数量进行判断:

1
2
3
4
5
6
7
# 获取图片集地址下的所有图片名称
image_names = [name for name in os.listdir(IMAGES_PATH) for item in IMAGES_FORMAT if
os.path.splitext(name)[1] == item]

# 简单的对于参数的设定和实际图片集的大小进行数量判断
if len(image_names) != IMAGE_ROW * IMAGE_COLUMN:
raise ValueError("合成图片的参数和要求的数量不能匹配!")

最后通过循环对图片进行拼接:

1
2
3
4
5
6
7
8
9
10
11
# 定义图像拼接函数
def image_compose():
to_image = Image.new('RGB', (IMAGE_COLUMN * IMAGE_WIDTH, IMAGE_ROW * IMAGE_HEIGHT)) #创建一个新图
# 循环遍历,把每张图片按顺序粘贴到对应位置上
for y in range(1, IMAGE_ROW + 1):
for x in range(1, IMAGE_COLUMN + 1):
from_image = Image.open(IMAGES_PATH + image_names[IMAGE_COLUMN * (y - 1) + x - 1]).resize(
(IMAGE_WIDTH, IMAGE_HEIGHT),Image.ANTIALIAS)
to_image.paste(from_image, ((x - 1) * IMAGE_WIDTH, (y - 1) * IMAGE_HEIGHT))
return to_image.save(IMAGE_SAVE_PATH) # 保存新图
image_compose() #调用函数

运行后拼接的图片如下图:

完整代码可以移步 Github 查看。

邹明远 wechat
欢迎关注我的公众号「Smart」
请我喝杯快乐水