import cv2
import os
from diffusers.utils import export_to_video
from PIL import Image

# 设置图片文件夹路径和输出视频文件路径

#image_folder = 'test/sample_2'
image_folder = 'test/sample_5'
output_video_path = './test/'

# 获取文件夹中的所有图片文件，并按文件名排序
images = [img for img in os.listdir(image_folder) if img.endswith(".png")]
#images=sorted(images,key=lambda x : int(x.split("_")[0]))
images=sorted(images,key=lambda x : int(x.split(".")[0]))

# 只取前49帧
#start_frame=39
#start_frame=24
#sample_5 start_frame=15
images = images[15:15+881]
print(images)

# 获取第一张图片的尺寸
first_image_path = os.path.join(image_folder, images[0])

first_image_path = os.path.join(image_folder, images[0])
first_image = Image.open(first_image_path)
width, height = first_image.size

# 计算新的尺寸，使其为16的倍数
new_width = (width + 15) // 16 * 16
new_height = (height + 15) // 16 * 16


frames = []
for image in images:
    img_path = os.path.join(image_folder, image)
    img = Image.open(img_path)

    # 调整图像尺寸
    resized_img = img.resize((new_width, new_height), Image.BILINEAR)
    frames.append(resized_img)

final_save_path=os.path.join(output_video_path, "sample_5_800.mp4")
export_to_video(frames, final_save_path, fps=16)