File size: 676 Bytes
e34aada |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
import os
def imgs_to_video(img_dir, video_path, audio_path=None, verbose=True):
cmd = f"ffmpeg -i {img_dir}/%5d.png "
if audio_path is not None:
cmd += f"-i {audio_path} "
cmd += "-strict -2 "
cmd += "-c:v libx264 -pix_fmt yuv420p -b:v 2000k -y -shortest "
if verbose is False:
cmd += " -v quiet "
cmd += f"{video_path} "
os.makedirs(os.path.dirname(video_path), exist_ok=True)
os.system(cmd)
if __name__ == '__main__':
imgs_to_video('infer_out/tmp_imgs', 'infer_out/tmp_imgs/out.mp4', 'data/raw/val_wavs/zozo.wav')
imgs_to_video('infer_out/tmp_imgs', 'infer_out/tmp_imgs/out2.mp4', 'data/raw/val_wavs/zozo.wav') |