Initial commit

03combine.py

@@ -0,0 +1,50 @@
+import os
+from math import ceil, sqrt
+from PIL import Image
+import random
+
+# 设置图片文件夹路径
+folder_path = 'output2'
+output_folder = 'output3'
+output_path = 'output3/merged_image.png'
+
+if not os.path.exists(output_folder):
+    os.makedirs(output_folder)
+
+# 获取所有 PNG 文件
+images = [os.path.join(folder_path, f) for f in os.listdir(folder_path) if f.lower().endswith('.png')]
+if not images:
+    print("没有找到 PNG 图片。")
+    exit()
+
+random.shuffle(images)
+
+num_images = len(images)
+# 计算行数和列数，使得接近正方形
+cols = ceil(sqrt(num_images))
+rows = ceil(num_images / cols)
+
+# 打开所有图片并获取尺寸
+opened_images = [Image.open(img) for img in images]
+widths, heights = zip(*(img.size for img in opened_images))
+
+# 假设所有图片尺寸相同，如果不同，可以选择调整大小
+max_width = max(widths)
+max_height = max(heights)
+
+# 创建一个新的空白图片
+merged_width = cols * max_width
+merged_height = rows * max_height
+merged_image = Image.new('RGBA', (merged_width, merged_height), (255, 255, 255, 0))
+
+# 将每张图片粘贴到合适的位置
+for index, img in enumerate(opened_images):
+    row = index // cols
+    col = index % cols
+    x = col * max_width
+    y = row * max_height
+    merged_image.paste(img, (x, y))
+
+# 保存合并后的图片
+merged_image.save(output_path)
+print(f"图片已成功合并并保存为 {output_path}")