import os
import imageio
from PIL import Image
from tqdm import tqdm   
from glob import glob
import numpy as np

def merge_images_to_video(path, name='train'):
    image_names = sorted(glob(os.path.join(path, '*.png')))
    video = imageio.get_writer(os.path.join(path, f"{name}_video.mp4"), mode='I', fps=5, codec='libx264',quality=10.0)
    
    for img_n in tqdm(image_names):
        img = np.array(Image.open(img_n))
        video.append_data(np.array(img))
    
    video.close()
    

if __name__ == '__main__':
    image_path = 'output/train/vis'
    merge_images_to_video(image_path)