(frames, save_path, fps, quality=9, ffmpeg_params=None)
| 237 | def save_video_ffmpeg(gen_video_samples, save_path, vocal_audio_list, fps=25, quality=5, high_quality_save=False): |
| 238 | |
| 239 | def save_video(frames, save_path, fps, quality=9, ffmpeg_params=None): |
| 240 | writer = imageio.get_writer( |
| 241 | save_path, fps=fps, quality=quality, ffmpeg_params=ffmpeg_params |
| 242 | ) |
| 243 | for frame in tqdm(frames, desc="Saving video"): |
| 244 | frame = np.array(frame) |
| 245 | writer.append_data(frame) |
| 246 | writer.close() |
| 247 | save_path_tmp = save_path + "-temp.mp4" |
| 248 | |
| 249 | if high_quality_save: |