我有一个GAN输出，我可以将下面输出序列中的图像保存为单个图像吗？

from PIL import Image
import os

def merge_images(image_folder, grid_size, output_filename):
    # 获取文件夹中的所有图像文件
    images = [Image.open(os.path.join(image_folder, img)) for img in os.listdir(image_folder)]
    
    # 计算每个图像的大小和最终合并图像的大小
    img_width, img_height = images[0].size
    grid_width = grid_size[0] * img_width
    grid_height = grid_size[1] * img_height
    
    # 创建一个新的空白图像
    merged_image = Image.new('RGB', (grid_width, grid_height))
    
    # 将每个图像放置到网格中的正确位置
    for idx, img in enumerate(images):
        x = (idx % grid_size[0]) * img_width
        y = (idx // grid_size[0]) * img_height
        merged_image.paste(img, (x, y))
    
    # 保存合并后的图像
    merged_image.save(output_filename)

# 使用示例
merge_images('path_to_your_image_folder', (4, 4), 'merged_output.jpg')