【实战教程】用CodeBuddy打造多风格图片合成神器

在美女图片的文件夹下存在多张图片,需求如下:
1.将文件夹下的图片合成一张图片
2.可以设置不同风格的图片, 给出选择

import os
import math
from PIL import Image
import random


def combine_images(input_folder, output_folder, base_filename="combined", max_width=2000, spacing=10):
    """
    将多张图片合成为一张图片，生成所有风格的合成图

    参数:
        input_folder: 输入图片文件夹路径
        output_folder: 输出文件夹路径
        base_filename: 输出文件基础名称
        max_width: 输出图片最大宽度
        spacing: 图片间距
    """
    # 获取所有图片文件
    image_files = []
    for f in os.listdir(input_folder):
        if f.lower().endswith(('.png', '.jpg', '.jpeg', '.gif', '.bmp')):
            image_files.append(os.path.join(input_folder, f))

    if not image_files:
        print("错误: 没有找到图片文件")
        return False

    # 打开所有图片并调整大小
    images = []
    for img_file in image_files:
        try:
            img = Image.open(img_file)
            # 统一调整为相同宽度，保持比例
            base_img_width = max_width // 4  # 基础宽度
            w_percent = (base_img_width / float(img.size[0]))
            h_size = int((float(img.size[1]) * float(w_percent)))
            img = img.resize((base_img_width, h_size), Image.LANCZOS)
            images.append(img)
        except Exception as e:
            print(f"警告: 无法处理图片 {img_file}: {e}")

    if not images:
        print("错误: 没有有效的图片可以处理")
        return False

    # 确保输出目录存在
    os.makedirs(output_folder, exist_ok=True)

    # 定义所有风格
    styles = {
        'grid': "网格布局",
        'vertical': "垂直布局",
        'horizontal': "水平布局",
        'collage': "艺术拼贴"
    }

    # 为每种风格生成合成图
    for style_code, style_name in styles.items():
        output_path = os.path.join(output_folder, f"{base_filename}_{style_code}.jpg")
        print(f"正在生成 {style_name} 风格的合成图...")

        if style_code == 'grid':
            # 网格布局
            cols = math.ceil(math.sqrt(len(images)))
            rows = math.ceil(len(images) / cols)
            img_width = images[0].width
            img_height = images[0].height

            total_width = cols * img_width + (cols - 1) * spacing
            total_height = rows * img_height + (rows - 1) * spacing

            result = Image.new('RGB', (total_width, total_height), (255, 255, 255))

            for i, img in enumerate(images):
                row = i // cols
                col = i % cols
                x = col * (img_width + spacing)
                y = row * (img_height + spacing)
                result.paste(img, (x, y))

        elif style_code == 'vertical':
            # 垂直布局
            total_width = max(img.width for img in images)
            total_height = sum(img.height for img in images) + (len(images) - 1) * spacing

            result = Image.new('RGB', (total_width, total_height), (255, 255, 255))

            y_offset = 0
            for img in images:
                result.paste(img, ((total_width - img.width) // 2, y_offset))
                y_offset += img.height + spacing

        elif style_code == 'horizontal':
            # 水平布局
            total_width = sum(img.width for img in images) + (len(images) - 1) * spacing
            total_height = max(img.height for img in images)

            result = Image.new('RGB', (total_width, total_height), (255, 255, 255))

            x_offset = 0
            for img in images:
                result.paste(img, (x_offset, (total_height - img.height) // 2))
                x_offset += img.width + spacing

        elif style_code == 'collage':
            # 艺术拼贴风格
            total_width = max_width
            total_height = int(total_width * 1.5)
            result = Image.new('RGB', (total_width, total_height), (250, 250, 250))

            for img in images:
                # 随机旋转角度
                angle = random.randint(-15, 15)
                rotated_img = img.rotate(angle, expand=True)

                # 随机位置
                x = random.randint(0, max(1, total_width - rotated_img.width))
                y = random.randint(0, max(1, total_height - rotated_img.height))

                # 随机透明度
                if random.random() > 0.7:
                    rotated_img = rotated_img.convert("RGBA")
                    rotated_img.putalpha(random.randint(150, 220))
                    result.paste(rotated_img, (x, y), rotated_img)
                else:
                    result.paste(rotated_img, (x, y))

        # 保存结果
        try:
            result.save(output_path, quality=95)
            print(f"成功: {style_name} 风格合成图已保存到 {output_path}")
        except Exception as e:
            print(f"错误: 保存 {style_name} 风格合成图失败: {e}")

    return True


if __name__ == "__main__":
    # 在这里设置你的输入和输出目录
    INPUT_FOLDER = "./美女图片"  # 替换为你的图片文件夹路径
    OUTPUT_FOLDER = "./合成结果"  # 替换为你想要的输出文件夹路径

    # 执行合成
    print("=== 开始图片合成 ===")
    combine_images(INPUT_FOLDER, OUTPUT_FOLDER)
    print("=== 图片合成完成 ===")