File size: 2,770 Bytes
f8a73ec |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
from PIL import Image
from typing import List
def create_image_grid(image_paths: List[str], output_path: str, grid_size: int = 2) -> None:
"""
将多个图片合并为 4 宫格图片，少于 4 个则留空，确保透明背景被填充为白色，并实现等比缩放。
参数:
- image_paths: 输入图片路径列表
- output_path: 输出 4 宫格图片路径
- grid_size: 网格大小（默认 2x2）
"""
images = []
target_size = (256, 256) # 每个格子的目标尺寸
for path in image_paths[:4]:
try:
# 加载图片，保留透明性
img = Image.open(path).convert('RGBA')
# 如果图片有透明通道，将透明区域填充为白色
if img.mode == 'RGBA':
background = Image.new('RGBA', img.size, (255, 255, 255, 255)) # 白色背景
img = Image.alpha_composite(background, img)
# 转换为 RGB 模式
img = img.convert('RGB')
original_width, original_height = img.size
aspect_ratio = original_width / original_height
if original_width >= original_height:
# 宽度是长边，调整宽度到 256
new_width = 256
new_height = int(256 / aspect_ratio)
else:
# 高度是长边，调整高度到 256
new_height = 256
new_width = int(256 * aspect_ratio)
# 等比缩放
img = img.resize((new_width, new_height), Image.Resampling.LANCZOS)
# img.thumbnail(target_size, Image.Resampling.LANCZOS) # 使用高质量缩放算法
# 创建 256x256 的空白画布（白色背景）
canvas = Image.new('RGB', target_size, (255, 255, 255))
# 计算居中位置
offset_x = (target_size[0] - img.size[0]) // 2
offset_y = (target_size[1] - img.size[1]) // 2
# 将缩放后的图片居中贴到画布上
canvas.paste(img, (offset_x, offset_y))
images.append(canvas)
except Exception as e:
print(f"Error loading image {path}: {e}")
images.append(None)
# 如果图片不足 4 张，补空
while len(images) < 4:
images.append(None)
# 创建空白画布（512x512，白色背景）
grid_image = Image.new('RGB', (512, 512), (255, 255, 255)) # 白色背景
# 按 2x2 排列贴图
for idx, img in enumerate(images):
if img is not None:
x = (idx % 2) * 256
y = (idx // 2) * 256
grid_image.paste(img, (x, y))
# 保存图片为 JPG 格式
grid_image.save(output_path, quality=95) # 设置质量为 95，避免过度压缩
|